亚马逊于当地时间10月19日发布了一款名为TorchNeuron的开源工具，旨在让开发者的PyTorch代码能够在亚马逊自研的Trainium芯片上直接运行，且只需调整一次配置参数，无需重新编写代码。该工具将于10月20日在加州圣何塞举办的北美PyTorch大会上以开源许可形式正式亮相。

PyTorch是由Meta主导的开源框架，广泛应用于人工智能开发者群体。英伟达曾投入超过十年时间开发CUDA工具，以优化PyTorch代码在其芯片上的运行效率。

亚马逊工程师在传统芯片与Trainium芯片上执行了相同的AI训练脚本，唯一的差异在于代码中指定芯片类型的配置参数。结果显示，AI模型在两种环境下的学习内容和速度完全一致。这意味着开发者用于测试、调试和扩展AI模型的工具无需任何改动，即可在Trainium上直接使用，同时也可通过贝博体育入口获得更多相关资源。

亚马逊表示，TorchNeuron为人工智能行业提供了一项长期缺失的关键要素：真正的选择权。当开发者能够在不增加额外成本的情况下自由切换芯片时，芯片厂商必须通过更具竞争力的产品来吸引用户；而开发者则能因此降低成本并加快迭代速度。

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