osayamenja / FlashMoELinks
Distributed MoE in a Single Kernel [NeurIPS '25]
☆190Updated this week
Alternatives and similar repositories for FlashMoE
Users that are interested in FlashMoE are comparing it to the libraries listed below
Sorting:
- ☆129Updated 5 months ago
- Building the Virtuous Cycle for AI-driven LLM Systems☆140Updated this week
- Autonomous GPU Kernel Generation via Deep Agents☆223Updated this week
- DeeperGEMM: crazy optimized version☆73Updated 8 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆141Updated 8 months ago
- ☆65Updated 9 months ago
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆158Updated 4 months ago
- High-performance distributed data shuffling (all-to-all) library for MoE training and inference☆109Updated last month
- Accelerating MoE with IO and Tile-aware Optimizations☆563Updated 2 weeks ago
- 16-fold memory access reduction with nearly no loss☆109Updated 10 months ago
- Utility scripts for PyTorch (e.g. Make Perfetto show some disappearing kernels, Memory profiler that understands more low-level allocatio…☆82Updated 4 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆160Updated 3 months ago
- ☆117Updated 8 months ago
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆165Updated 2 months ago
- Efficient Long-context Language Model Training by Core Attention Disaggregation☆85Updated this week
- ☆221Updated 2 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆319Updated this week
- Pipeline Parallelism Emulation and Visualization☆77Updated 3 weeks ago
- ☆35Updated 10 months ago
- Nex Venus Communication Library☆72Updated 2 months ago
- ☆342Updated this week
- ☆84Updated last year
- ByteCheckpoint: An Unified Checkpointing Library for LFMs☆268Updated last month
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆176Updated last year
- ☆102Updated last year
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆370Updated 6 months ago
- A lightweight design for computation-communication overlap.☆213Updated 2 weeks ago
- Sequence-level 1F1B schedule for LLMs.☆38Updated 5 months ago
- ☆60Updated last year
- nnScaler: Compiling DNN models for Parallel Training☆124Updated 4 months ago