GitHub repository: sustcsonglin/flash-linear-attention
Primary language: Python
Description: Efficient implementations of state-of-the-art linear attention models in Pytorch and Triton