亚马逊宣布推出开源工具 TorchNeuron，支持 PyTorch 代码在 Trainium 上运行

亚马逊宣布推出开源工具 TorchNeuron，使开发人员只需修改一次配置，即可让 PyTorch 代码在 Trainium 芯片上本地运行。亚马逊称，该工具计划于当地时间 10 月 20 日在北美 PyTorch 大会上以开源许可形式正式发布；同一训练脚本在传统芯片与 Trainium 上学习结果一致、学习速度相同。