flash-attn-from-scratch. NumPy implementation of FlashAttention from scratch - tiling, running softmax, and recomputation without a single GPU

github.com/AKMessi/flash-attn-from-scratch

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.