AWS发布TorchNeuron原生PyTorch后端支持Trainium

2026年10月19日，PyTorch官方宣布推出TorchNeuron，这是专为AWS Trainium芯片设计的开源原生PyTorch后端。该工具旨在简化开发者在AWS Trainium硬件上的模型部署流程。通过TorchNeuron，开发者只需将设备指向`.to('neuron')`或配置`torch.compile`后端，即可直接在Trainium2和Trainium3芯片上运行现有的模型代码、库及工具。这一更新意味着用户无需重写工作流或进行复杂的适配，便能利用Trainium芯片的算力优势，实现了从标准PyTorch环境到AWS专用AI加速器的无缝衔接。

AI 根据报道生成 · 2 小时前更新

还没有足够的连续观测数据，暂不绘制趋势。