https://github.com/daedalus/ssa

O(N·K) multi-head attention for PyTorch — a sparse drop-in replacement for dense scaled-dot-product attention.

attention-mechanism deep-learning efficient-computation long-context lsh machine-learning pytorch sparse-attention transformer

Last synced: 05 Jul 2026