亚马逊开源 TorchNeuron，支持 PyTorch 代码运行于 Trainium

亚马逊当地时间19日宣布推出开源工具 TorchNeuron，开发者无需重写 PyTorch 代码，仅需调整芯片配置，即可在亚马逊自研 Trainium 芯片上运行。该工具计划于20日在加州圣何塞举行的北美 PyTorch 大会上以开源许可发布。亚马逊工程师使用同一训练脚本分别在传统芯片和 Trainium 上测试，结果显示两种环境下模型训练内容和速度一致。亚马逊还表示正参与 PyTorch 上游开发，与社区合作移除框架中绑定特定芯片的代码，推动 PyTorch 更具硬件兼容性。