摩尔线程正式推出其开源大模型分布式训练仿真工具SimuMax的1.1版本。该版本在完整保留v1.0高保真仿真能力的前提下，实现了从独立仿真工具向一体化全栈工作流平台的关键跃迁，为大模型训练的仿真验证与性能调优提供了端到端的系统级支撑。

此次升级围绕三大核心突破展开：面向工程师的图形化配置界面、自动化并行策略探索引擎，以及融合计算与通信建模能力的System-Config自动生成流水线。同时，新版本显著增强了对主流训练框架Megatron-LM的适配能力，并进一步提升对混合并行（如数据并行+张量并行+专家并行）场景下复杂通信行为的建模保真度，使仿真结果更逼近真实集群运行状态。

SimuMax是一款专为大语言模型分布式训练负载打造的轻量级高精度仿真平台，支持从单卡设备至万卡级超大规集群的全尺度建模。它无需实际运行完整训练流程，即可精准预测显存占用、吞吐量、通信开销等关键指标，助力开发者深度理解训练瓶颈，高效定位性能优化路径。

据最新说明，SimuMax v1.1在延续原有高精度仿真优势的基础上，通过以下关键能力构建更完备、更智能的仿真工作流：

• 自适应并行策略搜索：集成启发式策略探索模块，可自动遍历多种并行组合及调度顺序，在多维约束下推荐最优执行方案，大幅降低人工调参门槛；
• 系统级配置生成流水线：新增基于硬件特征与通信拓扑的System-Config构建流程，同步建模计算效率与通信效率，实现更高保真的系统行为刻画；
• 框架兼容性增强与特性适配：正式支持Megatron-LM v0.14，重点兼容其新版MoE Router引入的显存压缩机制与动态路由逻辑；
• 细粒度带宽争用仿真：针对EP/TP/DP多重并行共存时的跨节点网络资源竞争现象，重构通信建模内核，显著提升万卡级集群中带宽瓶颈识别的准确性。

SimuMax v1.1全新上线交互式可视化配置面板。用户仅需通过拖拽、勾选与参数填写等简易操作，即可完成从单卡微调任务到万卡预训练作业的全流程定义，极大降低使用门槛，让研发人员得以聚焦于并行策略创新与底层性能挖掘，加速大模型训练范式的迭代演进。

源码地址：点击下载