品玩

科技创新者的每日必读

打开APP
关闭

摩尔线程发布 Torch-MUSA v2.7.0,强化大模型训练支持

5小时前

品玩11月28日讯,摩尔线程近日推出 PyTorch 框架 MUSA 扩展库 Torch-MUSA 的新版本 v2.7.0,持续完善基于国产全功能 GPU(MUSA 架构)的深度学习生态。

新版本新增动态双精度转换、分布式检查点异步保存等特性,支持 flash_attention、transformer_encoder_layer 等关键算子,专属算子总数超 1050 个;同时默认启用 TF32,并优化 torch.compile、FSDP2 内存占用及性能分析工具。此前 v2.5.0 已集成 muFFT、muSolver 加速库并支持统一内存管理。

该版本兼容最新 MUSA SDK 4.2.0+,未来将跟进 PyTorch 2.9.0,进一步提升训练与推理效率。

取消 发布

下载品玩App,比99.9%的人更先知道关于「摩尔线程」的新故事

下载品玩App

比99.9%的人更先知道关于「摩尔线程」的新故事

iOS版本 Android版本
立即下载
AI阅读助手
以下有两点提示,请您注意:
1. 请避免输入违反公序良俗、不安全或敏感的内容,模型可能无法回答不合适的问题。
2. 我们致力于提供高质量的大模型问答服务,但无法保证回答的准确性、时效性、全面性或适用性。在使用本服务时,您需要自行判断并承担风险;
感谢您的理解与配合
该功能目前正处于内测阶段,尚未对所有用户开放。如果您想快人一步体验产品的新功能,欢迎点击下面的按钮申请参与内测 申请内测