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

Brooooooklyn with Copilot wants to merge 1 commit into

flash-attnmlx-node/mlx:flash-attnfrom

copilot/sub-pr-1mlx-node/mlx:copilot/sub-pr-1Copy head branch name to clipboard