PyTorch· Nafea Bshara, Robert Hundt, Ron Diamant, Colin Brace, Rahul Solanki, Maen Suleiman, Khaled ElGalaind, Temesghen Kahsai, Cyril Bortolato, Indu Thangakrishnan, Shauheen Zahirazami, Sarang Ozarde, Adarsh Singh, Hanbo Wang, Esha Lakhotia, Jianying Lang·· 19 小时前精选AI 评分80
PyTorch 发布 TorchNeuron：AWS Trainium 的原生后端
Introducing TorchNeuron: Native PyTorch backend for AWS Trainium
AI 导读
PyTorch 宣布推出 TorchNeuron，这是专为 AWS Trainium 芯片设计的开源原生 PyTorch 后端。开发者只需将设备指向 `.to('neuron')` 或配置 `torch.compile` 后端，即可在 Trainium2 和 Trainium3 上直接运行现有的模型代码、库和工具，无需重写工作流。
推荐理由
TorchNeuron 作为开源原生后端，让现有 PyTorch 代码无需修改即可在 AWS Trainium 上运行，显著降低了硬件迁移的工程成本。
来源：PyTorch · pytorch.org