PyTorch 博客· The EUROPA consortium (Domyn, Fraunhofer IAIS and IIS)·· 3 小时前精选AI 评分65
EUROPA联盟使用PyTorch和TorchTitan在欧盟公共基础设施上训练1T参数MoE模型
Training a 1T-Parameter MoE on EU public infrastructure with PyTorch and TorchTitan
AI 导读
EUROPA联盟(由Domyn、Fraunhofer IAIS和IIS组成)在JUPITER超算上使用PyTorch和TorchTitan成功训练了约1万亿参数的MoE模型,通过FlexShard、激活检查点卸载至Grace CPU、FSDP流解耦等优化,将每GPU吞吐量从90提升至约1,300 tokens/s。
推荐理由
原文详细描述了在公共基础设施上训练千亿参数MoE模型的技术路径和优化细节,读者可据此理解大规模模型训练在非专属集群上的可行方案。
来源:PyTorch 博客 · pytorch.org