跳到正文
原文
AIHOT · AI 精选· noreply@aihot.news (PyTorch:Blog)·· 4 小时前精选AI 评分80

AWS 发布 TorchNeuron:Trainium 的原生 PyTorch 后端

AI 导读

AWS 宣布推出 TorchNeuron,这是一个面向 Trainium 芯片的开源原生 PyTorch 后端,将于 10 月 20 日以 Apache 2.0 协议公开 Beta。

推荐理由

原文给出了执行模式、分布式 API 和 100k 模型兼容性测试等细节,开发者可据此评估现有 PyTorch 代码迁移到 Trainium 的改动成本。

来源:AIHOT · AI 精选 · aihot.news