SoccerDream AI ← AI 圈资讯 产品发布

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

PyTorch:Blog(RSS) ·
AWS 发布 TorchNeuron:Trainium 的原生 PyTorch 后端

AWS 宣布 TorchNeuron,一个面向 Trainium 的开源原生 PyTorch 后端,10 月 20 日起以 Apache 2.0 公开 Beta 发布于 github.com/aws-neuron/torch-neuronx。

为什么值得看

原文给出了 TorchNeuron 的执行模式、分布式支持与实测覆盖情况,读者可以据此评估现有 PyTorch 工作流迁移到 Trainium 的成本。

阅读原文 · PyTorch:Blog(RSS)
本文由 SoccerDream AI 自动聚合索引,内容版权归原作者与 PyTorch:Blog(RSS) 所有。
本站仅作技术动态聚合,不构成任何形式的建议。