Skip to content

[Metal] Fused Flash Attention backward (VJP) kernels - #3241

Closed
Brooooooklyn wants to merge 1 commit into
ml-explore:mainfrom
mlx-node:flash-attn
Closed

[Metal] Fused Flash Attention backward (VJP) kernels#3241
Brooooooklyn wants to merge 1 commit into
ml-explore:mainfrom
mlx-node:flash-attn

Commits

  1. Commits on Mar 21, 2026