产品 · 精选
AWS 发布 TorchNeuron：Trainium 的原生 PyTorch 后端
导读
AWS 宣布 TorchNeuron，一个面向 Trainium 的开源原生 PyTorch 后端，10 月 20 日起以 Apache 2.0 公开 Beta 发布于 github.com/aws-neuron/torch-neuronx。
推荐理由
原文给出了 TorchNeuron 的执行模式、分布式支持与实测覆盖情况，读者可以据此评估现有 PyTorch 工作流迁移到 Trainium 的成本。