Distributed MoE in a Single Kernel [NeurIPS '25] - View it on GitHub