Torch-MLU – 寒武纪开源的PyTorch后端插件,支持大模型一键迁移

1个月前发布 51 00

Torch-MLU 是寒武纪开源的 PyTorch 设备后端扩展插件,支持开发者将寒武纪 MLU 系列智能加速卡作为 PyTorch 的加速后端使用。插件实现了对 PyTorch 的原生支持,开发者能无缝地将基于 GPU 的深度学习模型迁移到寒武纪 MLU 硬件上,提升模型的训练和推理效率。Torch-MLU 的开源进一步促进 AI 生态...

收录时间:
2025-12-16
Torch-MLU – 寒武纪开源的PyTorch后端插件,支持大模型一键迁移Torch-MLU – 寒武纪开源的PyTorch后端插件,支持大模型一键迁移

Torch-MLU 是寒武纪开源的 PyTorch 设备后端扩展插件,支持开发者将寒武纪 MLU 系列智能加速卡作为 PyTorch 的加速后端使用。插件实现了对 PyTorch 的原生支持,开发者能无缝地将基于 GPU 的深度学习模型迁移到寒武纪 MLU 硬件上,提升模型的训练和推理效率。Torch-MLU 的开源进一步促进 AI 生态的共建,为全球开发者提供更灵活和高效的开发环境。

官网:https://github.com/Cambricon/torch_mlu


立即打开官网

数据统计

相关导航