Posts tagged announce-transition-pytorch-trainium
Announcing Transition to PyTorch Native Support for AWS Trainium in the Next Neuron Release Supporting PyTorch 2.10
- 16 December 2025
This document is relevant for: Inf1, Inf2, Trn1, Trn2, Trn3
Starting with the introduction of Neuron support for PyTorch 2.10, AWS Neuron will begin a transition from PyTorch/XLA to native PyTorch support via TorchNeuron. PyTorch 2.9 will be the last version based on PyTorch/XLA.