# AWS 开放 TorchNeuron 公测，支持 Trainium 芯片

> AWS 发布 TorchNeuron 公测版。这款开源 PyTorch 后端可在 Trainium AI 芯片上原生运行。

Oossa · 2026-10-20 · https://oossa.com/zh/aws-opens-public-beta-of-torchneuron-for-trainium-chips

AWS 宣布，面向其 Trainium AI 芯片的原生 PyTorch 后端 TorchNeuron 已开启公测。这款开源库让开发者只需添加 .to('neuron')，即可在 Trainium 上运行 PyTorch 代码。它支持即时执行和 torch.compile 等标准 PyTorch 功能。开发者可通过 GitHub 仓库获取 TorchNeuron，并在 EC2 Trainium2 和 Trainium3 实例上使用。

## 事实

- 公测于 2026-10-20 开启
- TorchNeuron 可在 EC2 Trainium2 和 Trainium3 实例上运行

## 为什么重要

研究人员现在无需重写代码，就能在 AWS 高性能 Trainium 芯片上训练 PyTorch 模型，从而降低成本和工作量。

## 来源与参考

1. [AWS announces TorchNeuron: Native PyTorch Support for AWS Trainium (Public Beta)](https://aws.amazon.com/about-aws/whats-new/2026/10/aws-announce-native-pytorch-support-trainium) – AWS, 2026-10-20

最后更新: 2026-10-20