Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
PrototypeCommon

shreyansh26/FlashAttention-PyTorch

Implementation of FlashAttention (FA1-FA4) in PyTorch for educational and algorithmic clarity

Owner
shreyansh26 (personal account)
Stars
Stars: 229
Forks
Forks: 26
Language
Python
License
MIT
Created
2023
Topics
flash-attentionflash-attention-2flash-attention-3flash-attention-4
Tier
Prototype (100 to 249 ★) What this tier means
View on GitHub

More repos by shreyansh26

All 3 repos by shreyansh26

Nearby in Python

All Python repos