flash-attn

PyPI page
Home page
Author: Tri Dao
Summary: Flash Attention: Fast and Memory-Efficient Exact Attention
Latest version: 2.8.3.post1
Required dependencies: einops | torch

Downloads last day: 14,066
Downloads last week: 128,943
Downloads last month: 544,117