1. 首页
  2. 资讯
  3. 摩尔线程开源万卡级万亿参数MoE训练基础设施

摩尔线程开源万卡级万亿参数MoE训练基础设施

第一电动AI同学
10月28日,摩尔线程宣布开源大模型训练基础设施项目Megatron-MUSA-Titan。该项目基于摩尔线程全功能GPU,聚焦万卡级集群的万亿参数MoE模型训练需求,覆盖预训练与强化学习两大场景,为开发者提供面向MUSA平台的训练适配与系统优化能力。

据介绍,Megatron-MUSA-Titan已在摩尔线程万卡级集群的万亿参数MoE模型预训练中持续应用,经过长时间训练实践检验,实现高MFU与长稳训练,同时支持与slime、miles强化学习框架协同,结合Megatron训练与SGLang样本生成,构建贯通万亿参数MoE预训练与强化学习的训练基础。

在技术特性方面,Megatron-MUSA-Titan支持blockwiseFP8训练,将摩尔线程全功能GPU的低精度计算能力引入训练流程,提升大规模训练的计算效率。还可结合DeepEP与ACE异步通信引擎,加速MoE中的Token分发与结果合并,并实现专家并行与计算的通算重叠,让数据交换与计算任务协同执行,提升端到端训练效率。

来源:一电快讯

返回第一电动网首页 >

6点赞
发表评论
热文榜
第一电动网官方微信

反馈和建议 在线回复

您的询价信息
已经成功提交我们稍后会联系您进行报价!

第一电动网
Hello world!