近日,摩尔线程正式发布开源大模型分布式训练仿真工具SimuMax的1.1版本。该版本在完整继承v1.0高精度仿真能力的基础上,实现了从单一工具到一体化全栈工作流平台的重要升级,为大模型训练的仿真与调优提供系统化支持。

图示:SimuMax v1.1的可视化配置界面
快速开始:四步启动高效仿真
开发者可通过以下步骤,快速体验SimuMax:
克隆仓库:
·git clone git@github.com:MooreThreads/SimuMax.git
·cd SimuMax
安装Python包:
·pip install -r requirements.txt
·pip install -v -e .
运行示例:
参考项目中的教程和示例(如examples/perf_llama3_8b_tp1_pp2.py),即可开始使用SimuMax进行训练仿真。
启动SimuMax应用:
开发者可以选择启动SimuMax应用,访问交互式配置界面。
·cd app
·bash install.sh
·streamlit run streamlit_app.py
SimuMax已在GitHub全面开源,开发者可访问仓库获取源代码、详细文档和示例。摩尔线程鼓励开发者通过提交Issue报告问题或通过Pull Request贡献代码,共同促进SimuMax功能的完善和软件生态的繁荣。
·SimuMax GitHub开源地址:https://github.com/MooreThreads/SimuMax
·SimuMax v1.1版本:https://github.com/MooreThreads/SimuMax/releases/tag/v1.1
摩尔线程始终致力于为开发者提供高效、创新的软件工具链。SimuMax的发布,为大模型分布式训练提供了从可视化配置、自动化策略推荐到高精度仿真的完整工作流,助力AI产业提升算力利用率,探索更高效、更智能的训练范式。
未来,SimuMax团队将持续迭代,计划进一步扩展平台能力,包括支持ViT模型、引入大规模训练的数据并行(DP)衰减估计,以及支持计算通信重叠(TP/EP Overlap)模拟等功能,不断突破训练仿真优化的技术边界。
关于摩尔线程
摩尔线程以全功能GPU为核心,致力于向全球提供加速计算的基础设施和一站式解决方案,为各行各业的数智化转型提供强大的AI计算支持。我们的目标是成为具备国际竞争力的GPU领军企业,为融合人工智能和数字孪生的数智世界打造先进的加速计算平台。我们的愿景是为美好世界加速。
全部0条评论
快来发表一下你的评论吧 !