Skip to content

feat(attention): complete Flash Attention VJP implementation#1

Closed
Brooooooklyn wants to merge 1 commit into
mainfrom
flash-attn
Closed

feat(attention): complete Flash Attention VJP implementation#1
Brooooooklyn wants to merge 1 commit into
mainfrom
flash-attn

[Metal] Fused Flash Attention backward (VJP) kernels

985fdd0
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar