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

amorehead/jvp_flash_attention

Flash Attention Triton kernel with support for second-order derivatives

Owner
amorehead (personal account)
Stars
Stars: 183
Forks
Forks: 16
Language
Python
License
None
Created
2025
Tier
Prototype (100 to 249 ★) What this tier means
View on GitHub

More repos by amorehead

All 2 repos by amorehead

Nearby in Python

All Python repos