研究人员开发了 UBEP(统一总线专家并行),这是一个新的通信库,旨在优化大规模超级节点上的混合专家(MoE)模型。UBEP 解决了 MoE 通信中的关键瓶颈,包括执行序列化、同步开销和负载不平衡,这些在 NVIDIA 的 NVL72/576 和华为的 CloudMatrix384 等系统中普遍存在。实验表明,UBEP 可将 All-to-All 延迟最多降低 52.4%,并将 MoE 推理每输出令牌时间(TPOT)最多提高 11.1%。 AI
影响 优化大规模 MoE 模型的通信,可能在先进硬件上实现更高效的训练和推理。
排序理由 该集群包含一篇详细介绍用于优化 MoE 模型的新通信库的研究论文。
- All-to-all
- Bulk Synchronous Parallel
- CloudMatrix384
- Huawei
- Mixture-of-Experts
- NVIDIA
- NVL72/576
- Time Per Output Token
AI 生成摘要 · Google Gemini · 来自 2 个来源。 我们如何撰写摘要 →