Skip to content

Fix triangle_attention custom_vjp fwd/primal pytree mismatch (tuple vs list) - #295

Open
ssiddhantsharma wants to merge 1 commit into
NVIDIA:mainfrom
ssiddhantsharma:fix/triangle-attention-custom-vjp-list
Open

Fix triangle_attention custom_vjp fwd/primal pytree mismatch (tuple vs list)#295
ssiddhantsharma wants to merge 1 commit into
NVIDIA:mainfrom
ssiddhantsharma:fix/triangle-attention-custom-vjp-list

Fix triangle_attention custom_vjp fwd/primal pytree mismatch (tuple v…

4c1845d
Select commit
Loading
Failed to load commit list.
Sign in for the full log view