亚马逊推出开源工具 TorchNeuron，可让 PyTorch 代码在自研 Trainium 芯片运行

亚马逊推出开源工具 TorchNeuron，开发者只需更改一次配置无需重写，即可让原有 PyTorch 代码在其自研 Trainium 芯片上本地运行。该工具将在北美 PyTorch 大会上正式开源发布。实测表明相同 AI 训练脚本在传统芯片与 Trainium 芯片上运行效果一致，原有开发工具无需修改即可直接在 Trainium 上使用。该工具为行业带来芯片选择的自由度，可推动芯片厂商提升产品竞争力，帮助开发者降本提效。此外亚马逊正参与 PyTorch 上游开发，推动框架实现硬件无关性。