Posts tagged announce-transition-pytorch-trainium

Announcing Transition to PyTorch Native Support for AWS Trainium in the Next Neuron Release Supporting PyTorch 2.10

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.

If you are using PyTorch 2.9, it will be the last version of it that uses the PyTorch/XLA backend in Neuron.

Read more ...