/FlashAttention-PyTorch

Implementation of FlashAttention in PyTorch

Primary LanguagePythonMIT LicenseMIT

Watchers