挑战英伟达 CUDA：亚马逊推出开源工具 TorchNeuron，自研 Trainium 芯片轻松运行 PyTorch 代码 - LOL下注

当地时间10月19日，亚马逊对外发布了一款名为TorchNeuron的开源工具。通过这一工具，开发人员编写的PyTorch代码可以直接在亚马逊自研的Trainium芯片上运行，只需改动一次配置，无需重新编写代码。

TorchNeuron计划于当地时间10月20日在加州圣何塞举办的北美PyTorch大会上，以开源许可的形式正式亮相。

PyTorch是由Meta主导的开源框架，被大多数人工智能开发者广泛使用。英伟达花费了超过十年时间打造CUDA工具，使得PyTorch代码能够在其芯片上高效运行。

据亚马逊介绍，其工程师在传统芯片与Trainium芯片上执行了相同的人工智能训练脚本。唯一的差异在于代码中用于指定芯片类型的设置参数。结果显示，AI模型在两种环境下的学习内容和学习速度均保持一致。这意味着开发者原本用于测试、调试和扩展AI模型的工具，无需任何调整，即可直接应用于Trainium芯片。

亚马逊表示，TorchNeuron为人工智能行业提供了一项长期缺失的关键要素：真正可行的选择权。当开发者能够在不增加额外成本的情况下自由切换不同芯片时，芯片厂商必须通过提供更具竞争力的产品来赢得开发者的青睐；而开发者则能从中获益，实现成本降低和迭代速度的提升。

此外，亚马逊正积极参与PyTorch的上游开发工作，与PyTorch社区合作，移除那些将框架与特定芯片绑定的代码。其目标是让PyTorch对整个开发者社区而言更具“硬件无关性”，而不仅仅服务于Trainium用户。