flex-block-attn: an efficient block sparse attention computation library
☆132Dec 26, 2025Updated 8 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☆946Updated this week
- FA4-based Relative Attention Kernel developed by TML and Colfax☆18Sep 11, 2026Updated last week
- ☆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
- 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.
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- [ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.☆1,244Feb 25, 2026Updated 7 months ago
- ☆255Nov 19, 2025Updated 10 months ago
- SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable Sparse–Linear Attention☆545Feb 24, 2026Updated 7 months ago
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆204Feb 11, 2026Updated 7 months ago
- Distributed MoE in a Single Kernel [NeurIPS '25]☆291May 5, 2026Updated 4 months ago
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated last year
- Official repository Flash Local Linear Attention☆40May 28, 2026Updated 3 months ago
- DeepSeek-V3.2-Exp DSA Warmup Lightning Indexer training operator based on tilelang☆52Nov 19, 2025Updated 10 months 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.
- [ICML2025, NeurIPS2025 Spotlight] Sparse VideoGen 1 & 2: Accelerating Video Diffusion Transformers with Sparse Attention☆708Jul 4, 2026Updated 2 months ago
- Some funny cute/cuteDSL code snippets☆35Mar 2, 2026Updated 6 months ago
- Official implementation of paper "VMoBA: Mixture-of-Block Attention for Video Diffusion Models"☆65Jul 1, 2025Updated last year
- A plug-and-play compiler that delivers free-lunch optimizations for both inference and training.☆333Updated this week
- CUDA kernels for linear attention variants, written in CuTe DSL and CUTLASS C++.☆552Updated this week
- [CVPR 2026 Highlight] Official implementation of Log-linear Sparse Attention (LLSA).☆92May 1, 2026Updated 4 months ago
- A unified inference and post-training framework for accelerated video generation.☆4,501Updated 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,261Sep 1, 2026Updated 3 weeks 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.
- A curated list of recent papers on efficient video attention for video diffusion models, including sparsification, quantization, and cach…☆64Oct 27, 2025Updated 10 months ago
- ☆54May 19, 2025Updated last year
- Efficient Long-context Language Model Training by Core Attention Disaggregation☆105Apr 7, 2026Updated 5 months ago
- The evaluation framework for training-free sparse attention in LLMs☆132Jan 27, 2026Updated 7 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
- ☆39Dec 14, 2025Updated 9 months ago
- DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling☆43Updated this week
- NVFP4 Flash-Attention 4 on BlackWell☆66Updated this week
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Skills for writing tilelang and debugging with CUDA toolkits.☆144Sep 15, 2026Updated last week
- Distributed Compiler and Optimized Parallel Kernels☆1,553Updated this week
- NVIDIA cuTile learn☆168Dec 9, 2025Updated 9 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆770Aug 29, 2026Updated 3 weeks ago
- 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…☆154Updated this week