# TorchNeuron 为 AWS Trainium 提供原生 PyTorch 支持

> PyTorch 发布开源后端 TorchNeuron，让模型无需修改代码即可在 Trainium 芯片上运行。

Oossa · 2026-10-19 · https://oossa.com/zh/torchneuron-adds-native-pytorch-support-for-aws-trainium

PyTorch 宣布推出原生后端 TorchNeuron，可让 AWS Trainium 在 PyTorch 中显示为常规设备。借助这个新库，开发者只需使用 .to('neuron')，即可将模型切换到 Trainium；还可通过 torch.compile 提升生产环境中的运行速度。该库支持即时执行、自适应即时模式，以及通过 Neuron Kernel Interface 使用自定义内核。代码将以 Apache 2.0 许可协议开源，公开测试版将于 Oct. 20 推出。

## 事实

- TorchNeuron 计划于 Oct 20 开源发布（公开测试版）。
- 支持 Trainium2 和 Trainium3 芯片。

## 为什么重要

研究人员可以沿用现有 PyTorch 代码，利用 Trainium 的速度优势，无需重写工作流程。

## 来源与参考

1. [Introducing TorchNeuron: Native PyTorch backend for AWS Trainium](https://pytorch.org/blog/introducing-torchneuron-native-pytorch-backend-for-aws-trainium/) – PyTorch
2. [TorchNeuron: running HF models on AWS Trainium is easier than ever](https://huggingface.co/blog/hf-torchneuron) – Hugging Face, 2026-10-19
3. [Amazon is taking its biggest shot at Nvidia yet with new TorchNeuron dev tool](https://techcrunch.com/2026/10/19/amazon-is-taking-its-biggest-shot-at-nvidia-yet-with-new-torchneuron-dev-tool/) – TechCrunch, 2026-10-19

最后更新: 2026-10-19