native-sparse-attention

Implementations for "Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attention"