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分发与结果合并,并实现专家并行与计算的通算重叠,让数据交换与计算任务协同执行,提升端到端训练效率。
来源:一电快讯
返回第一电动网首页 >
以上内容由AI创作,如有问题请联系admin#d1ev.com(#替换成@)沟通,AI创作内容并不代表第一电动网(www.d1ev.com)立场。
文中图片源自互联网或AI创作,如有侵权请联系邮件删除。