摩尔线程Torch-MUSA插件v1.3.0开源升级,加速替代CUDA的步伐!

   发布时间:2024-10-31 21:20 作者:冯璃月

摩尔线程近日宣布,其针对PyTorch深度学习框架的MUSA插件“Torch-MUSA”已推出重大更新的v1.3.0版本,该版本全面兼容PyTorch 2.2.0,进一步提升了模型在摩尔线程GPU上的性能与覆盖度。

作为全球广泛应用的深度学习框架,PyTorch在自然语言处理、计算机视觉等领域具有强大的实力。摩尔线程的Torch-MUSA插件为PyTorch用户提供了MUSA后端加速,使得深度学习模型能够在MUSA架构上更加流畅地运行。

自Torch-MUSA首次发布以来,已经历了多个版本的迭代升级,不断优化兼容性与性能。从v1.0.0版本开始,该插件就率先支持了PyTorch 2.0,为用户带来了显著的计算加速与更好的易用性。

最新发布的v1.3.0版本除了全面支持PyTorch 2.2.0外,还极大地提升了模型训练与推理的效率,更好地满足了高性能深度学习任务的需求。该版本还增加了对FSDP等特性的支持,使得用户能够处理更复杂的模型和更大规模的数据。

目前,Torch-MUSA项目已完全开源,开发者可通过GitHub获取源代码,并积极参与项目的开发与改进。摩尔线程鼓励开发者通过提交问题报告或代码修改申请等方式,共同推动Torch-MUSA及MUSA软件生态的持续发展。

Torch-MUSA的功能特性包括轻松迁移现有的PyTorch模型到MUSA架构GPU上运行,完全兼容PyTorch的自动微分与动态图机制,并支持多种神经网络模块、优化算法及关键深度学习算子的加速。该插件还支持多种PyTorch特性,为用户提供了更为丰富的功能选择。

随着深度学习技术的不断发展,摩尔线程将继续跟进PyTorch的版本更新,并计划支持更高版本的PyTorch,以满足用户不断增长的需求。

 
 
更多>同类内容
全站最新
热门内容
本栏最新