flex-block-attn: an efficient block sparse attention computation library
☆130Dec 26, 2025Updated 7 months ago
Alternatives and similar repositories for flex-block-attn
Users that are interested in flex-block-attn are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A Distributed Attention Towards Linear Scalability for Ultra-Long Context, Heterogeneous Data Training☆910Updated this week
- FA4-based Relative Attention Kernel developed by TML and Colfax☆18Jul 17, 2026Updated 3 weeks ago
- ☆37Aug 7, 2025Updated last year
- High performance RMSNorm Implement by using SM Core Storage(Registers and Shared Memory)☆30Jan 22, 2026Updated 6 months ago
- ☆21Jun 12, 2025Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- [ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.☆1,027Feb 25, 2026Updated 5 months ago
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable Sparse–Linear Attention☆329Feb 24, 2026Updated 5 months ago
- ☆253Nov 19, 2025Updated 8 months ago
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆196Feb 11, 2026Updated 6 months ago
- Distributed MoE in a Single Kernel [NeurIPS '25]☆281May 5, 2026Updated 3 months ago
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated last year
- Official repository Flash Local Linear Attention☆38May 28, 2026Updated 2 months ago
- DeepSeek-V3.2-Exp DSA Warmup Lightning Indexer training operator based on tilelang☆50Nov 19, 2025Updated 8 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- [ICML2025, NeurIPS2025 Spotlight] Sparse VideoGen 1 & 2: Accelerating Video Diffusion Transformers with Sparse Attention☆701Jul 4, 2026Updated last month
- Some funny cute/cuteDSL code snippets☆33Mar 2, 2026Updated 5 months ago
- Official implementation of paper "VMoBA: Mixture-of-Block Attention for Video Diffusion Models"☆64Jul 1, 2025Updated last year
- A plug-and-play compiler that delivers free-lunch optimizations for both inference and training.☆326Updated this week
- CUDA kernels for linear attention variants, written in CuTe DSL and CUTLASS C++.☆538Updated this week
- A unified inference and post-training framework for accelerated video generation.☆3,943Updated this week
- [CVPR 2026 Highlight] Official implementation of Log-linear Sparse Attention (LLSA).☆93May 1, 2026Updated 3 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆134Jun 24, 2025Updated last year
- FlashKDA: high-performance Kimi Delta Attention kernels☆1,211Jul 30, 2026Updated 2 weeks ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A curated list of recent papers on efficient video attention for video diffusion models, including sparsification, quantization, and cach…☆61Oct 27, 2025Updated 9 months ago
- ☆54May 19, 2025Updated last year
- Efficient Long-context Language Model Training by Core Attention Disaggregation☆105Apr 7, 2026Updated 4 months ago
- The evaluation framework for training-free sparse attention in LLMs☆129Jan 27, 2026Updated 6 months ago
- NVFP4 Flash-Attention 4 on BlackWell☆41Jul 23, 2026Updated 3 weeks ago
- Sample Codes using NVSHMEM on Multi-GPU☆30Jan 22, 2023Updated 3 years ago
- Muon in Int8 Precision Made Possible☆20Jun 18, 2026Updated last month
- ☆40Dec 14, 2025Updated 8 months ago
- DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling☆36Aug 5, 2026Updated last 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.
- Skills for writing tilelang and debugging with CUDA toolkits.☆133May 20, 2026Updated 2 months ago
- Distributed Compiler based on Triton for Parallel Systems☆1,518Updated this week
- NVIDIA cuTile learn☆168Dec 9, 2025Updated 8 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆741Jul 4, 2026Updated last month
- Debug print operator for cudagraph debugging☆18Aug 2, 2024Updated 2 years ago
- ☆33Oct 15, 2025Updated 10 months ago
- MSLK (Meta Superintelligence Labs Kernels) is a collection of PyTorch GPU operator libraries that are designed and optimized for GenAI tr…☆145Updated this week