AWS 公测 TorchNeuron,开放 Trainium 原生 PyTorch 后端
AWS 于 10 月 20 日公测并开源 TorchNeuron,使 PyTorch 训练与推理代码可在 Trainium2、Trainium3 上原生运行。
2026 年 10 月 20 日,AWS 启动 TorchNeuron 公开测试并开放源代码。这是面向 Trainium 的原生 PyTorch 后端;开发者可通过 .to('neuron') 将现有代码移至 Trainium2 或 Trainium3 实例,并使用 eager execution、torch.compile 及标准 PyTorch API。它降低了在该芯片上试验训练和推理工作负载所需的代码适配量,目前状态为公开测试。
来源:AWS 发布说明 · TorchNeuron 代码仓库