小红书、北大开源 UltraEP:面向大规模 MoE 训推的「最优」负载均衡方案
AI 导读小红书与北大开源 UltraEP,首次将基于精确路由信息的实时负载均衡引入生产系统,在每个 microbatch 和每一层动态复制热点专家,平均达到理想性能的 94.3%,相比业界 SOTA 训推框架提升 1.49 倍。该方案已在 288B 参数 MoE 模型的完整预训练中部署,保持不低于理想性能 92% 的水平。
来源:公众号:小红书技术 · mp.weixin.qq.com
事件后续1
查看事件全部- 小红书、北大开源 UltraEP:面向大规模 MoE 训推的「最优」负载均衡方案公众号:小红书技术 · 7 月 20 日