osayamenja / FlashMoELinks
Distributed MoE in a Single Kernel [NeurIPS '25]
☆178Updated this week
Alternatives and similar repositories for FlashMoE
Users that are interested in FlashMoE are comparing it to the libraries listed below
Sorting:
- ☆126Updated 4 months ago
- DeeperGEMM: crazy optimized version☆74Updated 8 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆139Updated 7 months ago
- Building the Virtuous Cycle for AI-driven LLM Systems☆112Updated last week
- 16-fold memory access reduction with nearly no loss☆109Updated 9 months ago
- Autonomous GPU Kernel Generation via Deep Agents☆202Updated this week
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆160Updated 3 months ago
- ☆65Updated 8 months ago
- ☆216Updated last month
- ☆117Updated 7 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆522Updated last week
- ☆52Updated 7 months ago
- ☆35Updated 10 months ago
- nnScaler: Compiling DNN models for Parallel Training☆123Updated 3 months ago
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆159Updated 2 months ago
- Artifact for "Marconi: Prefix Caching for the Era of Hybrid LLMs" [MLSys '25 Outstanding Paper Award, Honorable Mention]☆48Updated 10 months ago
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆152Updated 3 months ago
- High-performance distributed data shuffling (all-to-all) library for MoE training and inference☆106Updated 2 weeks ago
- Pipeline Parallelism Emulation and Visualization☆74Updated this week
- Nex Venus Communication Library☆70Updated last month
- Implement Flash Attention using Cute.☆100Updated last year
- ☆133Updated 7 months ago
- ☆338Updated last week
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆364Updated 6 months ago
- [NeurIPS'25 Spotlight] Adaptive Attention Sparsity with Hierarchical Top-p Pruning☆83Updated last month
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆220Updated last year
- ☆100Updated last year
- A lightweight design for computation-communication overlap.☆209Updated 2 weeks ago
- Efficient Long-context Language Model Training by Core Attention Disaggregation☆75Updated 2 weeks ago
- Utility scripts for PyTorch (e.g. Make Perfetto show some disappearing kernels, Memory profiler that understands more low-level allocatio…☆76Updated 4 months ago