native-sparse-attention-pytorch. Implementation of the sparse attention pattern proposed by the Deepseek team in their "Native Sparse Attention" paper