Dist-MoE implements Speed-of-Light distributed Mixture-of-Experts forward and backward kernels. They are highly optimized and fused comm-compute kernels for training and inference on SM100+ GPUs
☆73Oct 29, 2026Updated last week
Alternatives and similar repositories for dist_moe
Users that are interested in dist_moe are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- High-performance RDMA for Python and PyTorch (ibverbs, efa, NVMe-oF)☆14Sep 2, 2026Updated 2 months ago
- Curated high-performance TIRx GPU kernels☆37Nov 4, 2026Updated last week
- ☆17Jun 15, 2026Updated 4 months ago
- mKernel: fast multi-node, multi-GPU fused kernels☆309Oct 17, 2026Updated 3 weeks ago
- ☆11Oct 29, 2026Updated last week
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A programmable distributed training system for PyTorch☆26Sep 22, 2026Updated last month
- ☆27Jan 18, 2026Updated 9 months ago
- Matrix multiplication on GPUs for matrices stored on a CPU. Similar to cublasXt, but ported to both NVIDIA and AMD GPUs.☆33Apr 2, 2025Updated last year
- Mixture-of-experts (MoE) training megakernel for NVL72s☆641Oct 1, 2026Updated last month
- ☆64Updated this week
- A cross-node expert-parallel MoE training library for computation-communication overlap.☆16Updated this week
- A lightweight CuTe-based CUDA kernel for DuoAttention, optimized for large language model inference.☆12Dec 25, 2025Updated 10 months ago
- Code for the paper ReplaySSM: Cache SSM Inputs, Not State☆69Oct 13, 2026Updated 3 weeks ago
- [ASPLOS' 26] TetriServe: Efficiently Serving Mixed DiT Workloads☆20Mar 12, 2026Updated 7 months ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- Winner 🏆 (Agent-only) MLSys 2026 - FlashInfer AI Kernel Generation Contest for the DeepSeek Sparse Attention (DSA) track with an average…☆318Sep 12, 2026Updated last month
- Communication patterns for AI, built on top of NCCL device and host APIs☆77Updated this week
- ☆47Aug 3, 2026Updated 3 months ago
- htop-like TUI for real-time RDMA network monitoring.☆129Updated this week
- Reverse-engineered instruction set architecture database for NVIDIA SM120 (Blackwell) GPUs.☆29Updated this week
- Reference implementation and examples of the CuTe Layout representation and algebra.☆395Nov 3, 2026Updated last week
- MoonEP: A Perfectly Balanced Expert Parallelism Library via Dynamic Redundant Experts☆1,210Oct 12, 2026Updated 3 weeks ago
- GPU Node ID (gni): hashes all GPU IDs in a node: `hash(GPU₀ + GPU₁ + ... + GPUₙ)`☆13Feb 20, 2026Updated 8 months ago
- Production-ready MoE load balancing via real-time expert replication☆263Oct 15, 2026Updated 3 weeks ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- DeepSelect: TopK kernels for DeepSeek Sparse Attention (DSA) and Samplers☆533Sep 30, 2026Updated last month
- ☆18May 27, 2026Updated 5 months ago
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆208Feb 11, 2026Updated 9 months ago
- ☆37Aug 7, 2025Updated last year
- Efficient Mask-Aware Distributed Attention for Long-Context Training☆62Updated this week
- TensorIR is a lightweight NVIDIA-owned MLIR compiler frontend for expressing tensor computations and lowering them to NVIDIA CUDA Tile IR…☆141Oct 28, 2026Updated 2 weeks ago
- ☆54Oct 22, 2026Updated 2 weeks ago
- Unified tracing profiler and visualizer, one timeline from CPU to GPU/HES to in-kernel zones☆152Sep 21, 2026Updated last month
- Ulysses sequence-parallel all-to-all as a torch custom op, moved by the GPU copy engines into torch symmetric memory. Zero SM usage; 1.66…☆54Aug 25, 2026Updated 2 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆68Feb 24, 2026Updated 8 months ago
- Bitwise-identical token probabilities between RL rollout (vLLM) and training☆38Updated this week
- DeDe (OSDI '25): an optimization framework for large-scale resource allocation☆15May 18, 2026Updated 5 months ago
- ☆64Sep 27, 2026Updated last month
- Paper recommendation bot using user profile to retrieve and recommend ArXiv papers, and post in Slack channels. User can use slack reacti…☆12Jan 21, 2026Updated 9 months ago
- ☆255Nov 19, 2025Updated 11 months ago
- ☆108Sep 8, 2026Updated 2 months ago
Are these results useful?