Moonshot open-sources MoonEP, a balanced MoE communication layer for GPUs and PPUs

deliprao · x · 2026-07-28

Moonshot’s MoonEP is an expert-parallelism communication library that keeps token loads balanced across ranks using dynamic redundant experts.

The repo describes online planning, a near-optimal GPU planning kernel with negligible overhead, and zero-copy/static-shape optimizations. It is designed to keep every rank receiving exactly S × K tokens even when routing is skewed, and it already targets NVIDIA GPUs while planning support for Alibaba’s TPU-like hardware.

Related event: MoonshotAI open-sources MoonEP for perfectly balanced expert parallelism(2 posts)→

Original post →

More from Infra

Infra channel →