# TorchNeuron adds native PyTorch support for AWS Trainium

> PyTorch releases TorchNeuron, an open‑source backend that lets models run on Trainium chips with unchanged code.

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

PyTorch announced TorchNeuron, a native backend that makes AWS Trainium appear as a regular PyTorch device. The new library lets developers switch a model to Trainium simply with .to('neuron') and use torch.compile for production speed. It supports eager execution, adaptive eager mode, and custom kernels via the Neuron Kernel Interface. The code will be open‑sourced under Apache 2.0 and the public beta starts on Oct. 20.

## The facts

- TorchNeuron open‑source release scheduled for Oct 20 (public beta).
- Runs on Trainium2 and Trainium3 chips.

## Why it matters

Researchers can use the same PyTorch code to leverage Trainium’s speed without rewriting their workflows.

## Sources & references

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

Last updated: 2026-10-19