🐳 Efficient Triton implementations for "Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attention"
☆1,021Feb 5, 2026Updated 7 months ago
Alternatives and similar repositories for native-sparse-attention
Users that are interested in native-sparse-attention are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Implementation of the sparse attention pattern proposed by the Deepseek team in their "Native Sparse Attention" paper☆815Aug 15, 2025Updated last year
- Efficient triton implementation of Native Sparse Attention.☆283May 23, 2025Updated last year
- 🚀 Efficient implementations for emerging model architectures☆5,771Updated this week
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆135Jun 24, 2025Updated last year
- MoBA: Mixture of Block Attention for Long-Context LLMs☆2,191Apr 3, 2025Updated last year
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Accelerating MoE with IO and Tile-aware Optimizations☆768Aug 29, 2026Updated 3 weeks ago
- A Distributed Attention Towards Linear Scalability for Ultra-Long Context, Heterogeneous Data Training☆945Updated this week
- ☆255Nov 19, 2025Updated 10 months ago
- qwen-nsa☆87Oct 14, 2025Updated 11 months ago
- 🔥 A minimal training framework for scaling FLA models☆415Apr 22, 2026Updated 4 months ago
- Distributed Compiler and Optimized Parallel Kernels☆1,547Updated this week
- A sparse attention kernel supporting mix sparse patterns☆555Jan 18, 2026Updated 8 months ago
- Muon is Scalable for LLM Training☆1,546Aug 3, 2025Updated last year
- ☆138May 29, 2025Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Helpful tools and examples for working with flex-attention☆1,248Updated this week
- Quantized Attention on GPU☆45Nov 22, 2024Updated last year
- [ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.☆1,165Feb 25, 2026Updated 6 months ago
- A Quirky Assortment of CuTe Kernels☆1,149Updated this week
- [ICLR 2025] DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Heads☆540Feb 10, 2025Updated last year
- DeeperGEMM: crazy optimized version☆85May 5, 2025Updated last year
- Ring attention implementation with flash attention☆1,057Sep 10, 2025Updated last year
- DeepSeek Native Sparse Attention pytorch implementation☆119Dec 17, 2025Updated 9 months ago
- FlashInfer: Kernel Library for LLM Serving☆6,460Updated this week
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆287Jun 6, 2025Updated last year
- ☆37Aug 7, 2025Updated last year
- [NeurIPS'24 Spotlight, ICLR'25, ICML'25] To speed up Long-context LLMs' inference, approximate and dynamic sparse calculate the attention…☆1,229Sep 10, 2026Updated last week
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆407Jul 10, 2025Updated last year
- A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention☆309Dec 1, 2025Updated 9 months ago
- Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels☆7,444Updated this week
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆266Aug 9, 2025Updated last year
- Compact and Agent-Native MoE Training System☆352Updated this week
- [MLSys'25] QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving; [MLSys'25] LServe: Efficient Long-sequence LLM Se…☆861Mar 6, 2025Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆65Apr 26, 2025Updated last year
- A fast communication-overlapping library for tensor/expert parallelism on GPUs.☆1,363Aug 28, 2025Updated last year
- Efficient Triton Kernels for LLM Training☆6,620Updated this week
- Tile primitives for speedy kernels☆3,721Sep 12, 2026Updated last week
- ☆54May 19, 2025Updated last year
- ☆125May 19, 2025Updated last year
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆693Updated this week