osayamenja / FlashMoELinks
Distributed MoE in a Single Kernel [NeurIPS '25]
☆49Updated this week
Alternatives and similar repositories for FlashMoE
Users that are interested in FlashMoE are comparing it to the libraries listed below
Sorting:
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆130Updated 2 weeks ago
- A lightweight design for computation-communication overlap.☆177Updated last week
- ☆83Updated 2 years ago
- ☆151Updated last year
- nnScaler: Compiling DNN models for Parallel Training☆118Updated last week
- ☆121Updated 9 months ago
- A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores☆53Updated last year
- ☆75Updated 4 years ago
- Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.☆66Updated 6 months ago
- LLM serving cluster simulator☆114Updated last year
- Github mirror of trition-lang/triton repo.☆78Updated this week
- ☆72Updated last year
- High performance Transformer implementation in C++.☆134Updated 8 months ago
- DeeperGEMM: crazy optimized version☆71Updated 4 months ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆122Updated 3 years ago
- Tile-based language built for AI computation across all scales☆61Updated this week
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆80Updated 10 months ago
- SpInfer: Leveraging Low-Level Sparsity for Efficient Large Language Model Inference on GPUs☆58Updated 6 months ago
- ☆85Updated 6 months ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆138Updated 2 years ago
- A baseline repository of Auto-Parallelism in Training Neural Networks☆146Updated 3 years ago
- ☆237Updated last year
- kvcached: Elastic KV cache for dynamic GPU sharing and efficient multi-LLM inference.☆94Updated this week
- Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning☆23Updated 4 months ago
- ☆144Updated 4 months ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆51Updated last year
- ☆123Updated 10 months ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆221Updated 2 years ago
- ☆57Updated 4 months ago
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆186Updated 8 months ago