/FlashAttention20Triton

Triton implementation of Flash Attention2.0

Primary LanguagePythonMIT LicenseMIT

Watchers