亚马逊推出TorchNeuron,让AI开发者无需改写代码即可在Trainium芯片上运行PyTorch
由 WORLDTECH 译自英文。事实、数字、名称和来源均与英文原文一致。 English

事件经过
亚马逊旗下的Annapurna Labs部门于2026年10月19日宣布推出TorchNeuron,公开测试版于次日在PyTorch大会上发布。该软件采用Apache 2.0开源许可证,是一个与PyTorch直接集成的后端,使Trainium芯片(具体为Trainium2和Trainium3系列)在框架看来如同原生设备。
实际效果是,一支在英伟达硬件上运行工作负载的团队,理论上无需改动代码库即可迁移到AWS的Trainium基础设施。2026年24日,亚马逊希望让更换英伟达芯片的复杂程度,大致相当于修改一个配置文件中的设置。NVIDIA是一家总部位于圣克拉拉的半导体公司,其产品和服务包括图形处理器及软件。
这并非夸大其词:该公司的新TorchNeuron软件后端让开发者只需改动一行代码,即可将现有PyTorch代码迁移到亚马逊的Trainium AI芯片上,把.to('cuda')改为.to('neuron'),无需重写或重新验证。选择这一时机是有意为之:PyTorch是AI模型开发中占主导地位的框架,在其旗舰大会上亮相,等于直接在英伟达的地盘上插上了旗帜。
在正式发布前,有三十多家早期采用者参与了alpha和beta测试。
这些企业包括语音AI公司Cartesia和Deepgram,以及利伯脑发育研究所,后者一直将该工具用于药物研究。早期用户表示切换后节省了成本,但亚马逊尚未公布具体数字。
一份每日简报,汇集AI、科技及其推动的市场动态。加入加密货币、金融和政策领域的34000多名读者行列。