亚马逊开源TorchNeuron 仅需改一行配置即可在Trainium上运行PyTorch

亚马逊于10月19日宣布推出开源工具TorchNeuron，该工具允许开发者仅通过改一处配置，就能在自研的Trainium芯片上原生运行PyTorch代码。

亚马逊工程师通过实验证明，同一份AI训练脚本在传统芯片和Trainium上运行，仅需修改指定芯片的设置参数，模型学习内容和速度完全一致。

TorchNeuron的推出旨在打破英伟达CUDA生态的垄断。亚马逊希望通过提供硬件无关的开发环境，让开发者能自由切换芯片而不付出额外代价。

亚马逊正主动参与PyTorch上游开发，与社区协作移除将框架与特定芯片绑定的代码，目标是让PyTorch对所有开发者更加硬件无关。

在算力争夺白热化的当下，亚马逊这一步既是为自家Trainium芯片铺路，也是借开源松动英伟达的生态垄断。