GitHub repository: 01-ai/flash-attention
Description: Fast and memory-efficient exact attention