AWS发布TorchNeuron,PyTorch应用可单行配置迁移至Trainium
亚马逊称TorchNeuron无需改写代码,并正将相关代码回馈上游PyTorch项目,以解除框架与特定芯片的绑定。
据techcrunch.com报道,亚马逊云计算服务(AWS)于2026年10月19日发布名为TorchNeuron的工具,开发者只需修改一行配置,即可让基于PyTorch编写的AI应用在亚马逊自研的Trainium芯片上运行。
亚马逊表示,TorchNeuron无需重写代码,熟悉PyTorch的开发者可以直接上手使用。AWS称,Trainium在同一颗芯片上同时承担训练与推理任务,并与AWS云基础设施集成。
亚马逊还表示,正在将相关代码回馈至PyTorch主项目,以移除把该框架与特定芯片绑定的代码。
亚马逊称,约85%的AI应用使用PyTorch编写。AWS安纳波利斯实验室技术软件副总裁Colin Brace就此次发布作出说明。
长期由英伟达主导的软件适配
PyTorch最初由Facebook开发,自2022年起归属Linux基金会,由该基金会维护开源项目。多年来,英伟达持续确保其CUDA软件支持用PyTorch编写的AI应用。
快速问答
TorchNeuron需要重写代码吗?
亚马逊表示不需要,只需修改一行配置,熟悉PyTorch的开发者可直接使用。
TorchNeuron运行在哪种芯片上?
运行在亚马逊自研的Trainium芯片上,该芯片同时承担训练与推理任务。