viai957/Flash-Attention-101

Flash Attention from First Principles: Triton & CUDA implementations with handwritten derivations, notebooks, and Colab benchmarks comparing PyTorch and Triton versions.

语言：Jupyter Notebook · GitHub 星标：9 · Fork：1

在 GitHub 上查看原仓库