GitHub repository: xiayuqing0622/flex_head_fa
Primary language: Python
Description: Fast and memory-efficient exact attention