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