Implementation of Flash Attention in Jax
Primary LanguagePythonMIT LicenseMIT
No one’s watching this repository yet.