GitHub repository: Dao-AILab/flash-attention
Primary language: Python
Description: Fast and memory-efficient exact attention