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

标题：亚马逊推出TorchNeuron挑战英伟达CUDA

10月19日，亚马逊宣布推出开源工具TorchNeuron。借助该工具，开发者编写的PyTorch代码可在亚马逊自研Trainium芯片上本地运行，只需更改一次配置，无需重写。该工具将于10月20日在加州圣何塞举行的北美PyTorch大会上以开源许可形式正式发布。

PyTorch由Meta主导，是多数人工智能开发者采用的开源框架。英伟达投入十多年研发CUDA工具，使PyTorch代码能在其芯片上高效运行。亚马逊此次推出的TorchNeuron，被视为对CUDA生态的直接挑战。

亚马逊工程师在传统芯片和Trainium芯片上运行了相同的AI训练脚本。唯一的区别是代码中指定所用芯片的设置参数。结果显示，AI模型在两种环境下学习到的内容完全一致，学习速度也相同。

这意味着开发者原本用于测试、调试和扩展AI模型的工具，无需任何修改即可直接在Trainium上使用。亚马逊称，TorchNeuron为人工智能行业带来一项长期缺失的关键要素：真正可行的选择权。当开发者能在不同芯片间自由切换而无需付出额外代价时，芯片厂商就必须以更具竞争力的产品赢得青睐。

对开发者而言，这种切换能力有望降低成本和提升迭代速度。对芯片厂商而言，竞争压力将更多落在产品性能和生态建设上。亚马逊此举也将其自研芯片进一步推向主流AI开发流程。

此外，亚马逊正积极参与PyTorch上游开发，与PyTorch社区合作移除将框架与特定芯片相绑定的代码。其目标是让PyTorch对整个开发者社区更具“硬件无关性”，而不仅服务于Trainium用户。这一方向能否改变现有芯片生态，仍有待开发者实践检验。