Meta 开源 Dist-MoE:解决 CUDA Graph 下的无 dropout MoE 难题

ezyang · x · 2026-10-21

Meta 超级智能实验室(MSL)开源 Dist-MoE,一个面向分布式 MoE 训练与推理的极速内核库。PyTorch 核心开发者 ezyang 表示,去年团队还在为「可 CUDA Graph 化、无 dropout 的 MoE」发愁,Dist-MoE 已在内部和开源侧基本解决了这个问题。

技术要点:

官方还配了图解指南,可交互查看路由、expert parallelism 与 fused kernel 的执行过程,是 MoE 分布式训练工程的优质参考资料。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →