flash-sparse-attention:基于 Flash Attention 的可训练稀疏注意力实现项目

Trainable fast and memory-efficient sparse attention

分支25Tags32