Torch

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