Fast and memory-efficient exact kmeans
☆140Feb 18, 2026Updated 2 weeks ago
Alternatives and similar repositories for flash-kmeans
Users that are interested in flash-kmeans are comparing it to the libraries listed below
Sorting:
- An experimental communicating attention kernel based on DeepEP.☆35Jul 29, 2025Updated 7 months ago
- ☆36Sep 6, 2025Updated 6 months ago
- Xmixers: A collection of SOTA efficient token/channel mixers☆28Sep 4, 2025Updated 6 months ago
- ☆52May 19, 2025Updated 9 months ago
- Fast and memory-efficient exact attention☆18Updated this week
- [ICML2025, NeurIPS2025 Spotlight] Sparse VideoGen 1 & 2: Accelerating Video Diffusion Transformers with Sparse Attention☆633Updated this week
- Benchmark tests supporting the TiledCUDA library.☆18Nov 19, 2024Updated last year
- ☆65Apr 26, 2025Updated 10 months ago
- Implementation for FP8/INT8 Rollout for RL training without performence drop.☆293Nov 7, 2025Updated 4 months ago
- Debug print operator for cudagraph debugging☆14Aug 2, 2024Updated last year
- Quantized Attention on GPU☆44Nov 22, 2024Updated last year
- DeeperGEMM: crazy optimized version☆74May 5, 2025Updated 10 months ago
- mHC-lite: You Don’t Need 20 Sinkhorn-Knopp Iterations☆70Jan 12, 2026Updated last month
- ☆118May 19, 2025Updated 9 months ago
- ☆227Nov 19, 2025Updated 3 months ago
- Distributed Compiler based on Triton for Parallel Systems☆1,371Feb 13, 2026Updated 3 weeks ago
- [WIP] Better (FP8) attention for Hopper☆32Feb 24, 2025Updated last year
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆239Jun 15, 2025Updated 8 months ago
- ☆88Updated this week
- [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling☆40Dec 2, 2023Updated 2 years ago
- Transformers components but in Triton☆34May 9, 2025Updated 9 months ago
- ☆38Aug 7, 2025Updated 7 months ago
- patches for huggingface transformers to save memory☆35Jun 2, 2025Updated 9 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆129Jun 24, 2025Updated 8 months ago
- A collection of specialized agent skills for AI infrastructure development, enabling Claude Code to write, optimize, and debug high-perfo…☆88Feb 2, 2026Updated last month
- Accelerating MoE with IO and Tile-aware Optimizations☆597Feb 27, 2026Updated last week
- ☆135Aug 18, 2025Updated 6 months ago
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆374Jul 10, 2025Updated 7 months ago
- Accelerate LLM preference tuning via prefix sharing with a single line of code☆51Jul 4, 2025Updated 8 months ago
- Official repository for the paper Local Linear Attention: An Optimal Interpolation of Linear and Softmax Attention For Test-Time Regressi…☆23Oct 1, 2025Updated 5 months ago
- A Distributed Attention Towards Linear Scalability for Ultra-Long Context, Heterogeneous Data Training☆659Updated this week
- flex-block-attn: an efficient block sparse attention computation library☆124Dec 26, 2025Updated 2 months ago
- [NeurIPS'25 Spotlight] Adaptive Attention Sparsity with Hierarchical Top-p Pruning☆88Nov 29, 2025Updated 3 months ago
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆91Jul 17, 2025Updated 7 months ago
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Jun 6, 2024Updated last year
- [ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.☆954Feb 25, 2026Updated last week
- [ICLR'25] Fast Inference of MoE Models with CPU-GPU Orchestration☆262Nov 18, 2024Updated last year
- Image Tokenizer Needs Post-Training☆24Oct 4, 2025Updated 5 months ago
- A Benchmark and Evaluation Suite for Zero-shot Singing Voice Synthesis☆23Feb 11, 2026Updated 3 weeks ago