flex-block-attn: an efficient block sparse attention computation library
☆133Dec 26, 2025Updated 10 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:
- FA4-based Relative Attention Kernel developed by TML and Colfax☆18Sep 11, 2026Updated last month
- A Distributed Attention Towards Linear Scalability for Ultra-Long Context, Heterogeneous Data Training☆967Updated this week
- ☆37Aug 7, 2025Updated last year
- CUDA Graph runtime tensor debugging utilities for PyTorch☆18Jul 12, 2026Updated 3 months ago
- High performance RMSNorm Implement by using SM Core Storage(Registers and Shared Memory)☆31Jan 22, 2026Updated 9 months ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- The code implementation of paper "VORTA: Efficient Video Diffusion via Routing Sparse Attention"☆12Oct 15, 2025Updated last year
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- ☆54May 19, 2025Updated last year
- DeepSeek-V3.2-Exp DSA Warmup Lightning Indexer training operator based on tilelang☆52Nov 19, 2025Updated 11 months ago
- ☆255Nov 19, 2025Updated 11 months ago
- ☆24Jun 12, 2025Updated last year
- [ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.☆1,266Feb 25, 2026Updated 8 months ago
- CUDA kernels for linear attention variants, written in CuTe DSL and CUTLASS C++.☆561Updated this week
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆208Feb 11, 2026Updated 8 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ☆11Updated this week
- NVFP4 Flash-Attention 4 on BlackWell☆69Sep 26, 2026Updated last month
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated last year
- [ICML2025, NeurIPS2025 Spotlight] Sparse VideoGen 1 & 2: Accelerating Video Diffusion Transformers with Sparse Attention☆712Jul 4, 2026Updated 4 months ago
- A lightweight CuTe-based CUDA kernel for DuoAttention, optimized for large language model inference.☆12Dec 25, 2025Updated 10 months ago
- Distributed MoE in a Single Kernel [NeurIPS '25]☆330May 5, 2026Updated 5 months ago
- Official implementation of paper "VMoBA: Mixture-of-Block Attention for Video Diffusion Models"☆65Jul 1, 2025Updated last year
- Official repository Flash Local Linear Attention☆40May 28, 2026Updated 5 months ago
- ☆15Jul 10, 2026Updated 3 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.
- Some funny cute/cuteDSL code snippets☆35Mar 2, 2026Updated 8 months ago
- A plug-and-play compiler that delivers free-lunch optimizations for both inference and training.☆335Updated this week
- [CVPR 2026 Highlight] Official implementation of Log-linear Sparse Attention (LLSA).☆94May 1, 2026Updated 6 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆135Jun 24, 2025Updated last year
- ☆20May 30, 2024Updated 2 years ago
- ☆16Jul 19, 2026Updated 3 months ago
- SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable Sparse–Linear Attention☆552Oct 18, 2026Updated 2 weeks ago
- Skills for writing tilelang and debugging with CUDA toolkits.☆149Oct 23, 2026Updated last week
- Efficient Long-context Language Model Training by Core Attention Disaggregation☆106Apr 7, 2026Updated 6 months ago
- 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.
- Muon in Int8 Precision Made Possible☆20Jun 18, 2026Updated 4 months ago
- W4A16 MoE GEMM CUDA kernel. Tuned for H200, B200 & B300.☆14Updated this week
- ☆118Updated this week
- The evaluation framework for training-free sparse attention in LLMs☆134Jan 27, 2026Updated 9 months ago
- A curated list of recent papers on efficient video attention for video diffusion models, including sparsification, quantization, and cach…☆69Oct 27, 2025Updated last year
- ☆40Dec 14, 2025Updated 10 months ago
- Sample Codes using NVSHMEM on Multi-GPU☆30Jan 22, 2023Updated 3 years ago