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