flex-block-attn: an efficient block sparse attention computation library
☆132Dec 26, 2025Updated 9 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☆952Updated this week
- FA4-based Relative Attention Kernel developed by TML and Colfax☆18Sep 11, 2026Updated last month
- ☆37Aug 7, 2025Updated last year
- High performance RMSNorm Implement by using SM Core Storage(Registers and Shared Memory)☆31Jan 22, 2026Updated 8 months ago
- ☆23Jun 12, 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.
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- [ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.☆1,260Feb 25, 2026Updated 7 months ago
- ☆255Nov 19, 2025Updated 10 months ago
- SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable Sparse–Linear Attention☆550Feb 24, 2026Updated 7 months ago
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆206Feb 11, 2026Updated 8 months ago
- Distributed MoE in a Single Kernel [NeurIPS '25]☆310May 5, 2026Updated 5 months ago
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated last year
- Official repository Flash Local Linear Attention☆40May 28, 2026Updated 4 months ago
- DeepSeek-V3.2-Exp DSA Warmup Lightning Indexer training operator based on tilelang☆52Nov 19, 2025Updated 10 months ago
- 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.
- [ICML2025, NeurIPS2025 Spotlight] Sparse VideoGen 1 & 2: Accelerating Video Diffusion Transformers with Sparse Attention☆709Jul 4, 2026Updated 3 months ago
- Some funny cute/cuteDSL code snippets☆35Mar 2, 2026Updated 7 months ago
- Official implementation of paper "VMoBA: Mixture-of-Block Attention for Video Diffusion Models"☆65Jul 1, 2025Updated last year
- CUDA kernels for linear attention variants, written in CuTe DSL and CUTLASS C++.☆555Updated this week
- A plug-and-play compiler that delivers free-lunch optimizations for both inference and training.☆334Oct 1, 2026Updated last week
- [CVPR 2026 Highlight] Official implementation of Log-linear Sparse Attention (LLSA).☆93May 1, 2026Updated 5 months ago
- A unified inference and post-training framework for accelerated video generation.☆4,596Updated this week
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆135Jun 24, 2025Updated last year
- FlashKDA: high-performance Kimi Delta Attention kernels☆1,280Sep 1, 2026Updated last month
- 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.
- A curated list of recent papers on efficient video attention for video diffusion models, including sparsification, quantization, and cach…☆65Oct 27, 2025Updated 11 months ago
- ☆54May 19, 2025Updated last year
- The evaluation framework for training-free sparse attention in LLMs☆133Jan 27, 2026Updated 8 months ago
- Efficient Long-context Language Model Training by Core Attention Disaggregation☆106Apr 7, 2026Updated 6 months ago
- Sample Codes using NVSHMEM on Multi-GPU☆30Jan 22, 2023Updated 3 years ago
- Muon in Int8 Precision Made Possible☆20Jun 18, 2026Updated 3 months ago
- ☆40Dec 14, 2025Updated 10 months ago
- DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling☆45Oct 5, 2026Updated last week
- NVFP4 Flash-Attention 4 on BlackWell☆68Sep 26, 2026Updated 2 weeks ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Skills for writing tilelang and debugging with CUDA toolkits.☆145Sep 15, 2026Updated 3 weeks ago
- Distributed Compiler and Optimized Parallel Kernels☆1,566Sep 18, 2026Updated 3 weeks ago
- NVIDIA cuTile learn☆168Dec 9, 2025Updated 10 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆779Aug 29, 2026Updated last month
- Debug print operator for cudagraph debugging☆18Aug 2, 2024Updated 2 years ago
- ☆33Oct 15, 2025Updated 11 months ago
- MSLK (Meta Superintelligence Labs Kernels) is a collection of PyTorch GPU operator libraries that are designed and optimized for GenAI tr…☆155Updated this week