Official code for the paper Hierarchical Sparse Attention Done Right: Toward Infinite Context Modeling. HiLS-Attention is a chunk-wise sparse attention mechanism that learns chunk selection end-to-end ...