lucidrains/flash-attention-jax - Gitstar Ranking

Gitstar Ranking

Users
Organizations
Repositories
Rankings

Sign in with GitHub

lucidrains

Fetched on 2025/01/09 06:29

lucidrains / flash-attention-jax

Implementation of Flash Attention in Jax - View it on GitHub

Star

205

Rank

143621

Released by @k0kubun in December 2014. Fork me on GitHub.