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