osayamenja / FlashMoELinks
Distributed MoE in a Single Kernel [NeurIPS '25]
☆89Updated 3 weeks ago
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☆138Updated last month
- A lightweight design for computation-communication overlap.☆181Updated last week
- ☆152Updated last year
- ☆141Updated 9 months ago
- ☆83Updated 2 years ago
- nnScaler: Compiling DNN models for Parallel Training☆118Updated 3 weeks ago
- Github mirror of trition-lang/triton repo.☆86Updated this week
- A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores☆53Updated last year
- TVM FFI☆79Updated this week
- ☆57Updated 4 months ago
- ☆150Updated 5 months ago
- ASPLOS'24: Optimal Kernel Orchestration for Tensor Programs with Korch☆37Updated 6 months ago
- ☆240Updated last year
- 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
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆100Updated 3 months ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆51Updated last year
- ☆90Updated 6 months ago
- ☆74Updated this week
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆122Updated 3 years ago
- DeeperGEMM: crazy optimized version☆72Updated 5 months ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆221Updated 2 years ago
- ☆75Updated 4 years ago
- Tile-based language built for AI computation across all scales☆67Updated 3 weeks ago
- ☆107Updated 5 months ago
- [NeurIPS 2025] ClusterFusion: Expanding Operator Fusion Scope for LLM Inference via Cluster-Level Collective Primitive☆45Updated 3 weeks ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆80Updated 11 months ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆138Updated 2 years ago
- A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.☆60Updated last week
- ☆92Updated 11 months ago
- High performance Transformer implementation in C++.☆135Updated 9 months ago