Skip to content
What are you looking for?. Open searchSR

Tags

PyTorch

1 item
Amazon's diagram of four stacked layers: PyTorch code moving a model to the neuron device, TorchNeuron turning it into ATen operations, the AWS Neuron compiler producing StableHLO, and a Trainium chip running the compiled binary.
Toolsmedium signal

Amazon's TorchNeuron, now in public beta, runs PyTorch code on Trainium chips as a standard device

Amazon announced TorchNeuron on October 19, 2026, a native PyTorch backend for its Trainium2 and Trainium3 chips. The public beta, released as open source under the Apache 2.0 license, followed a day later. Amazon says existing PyTorch code runs on Trainium with one configuration change, but the first release leaves out several PyTorch features, and no independent performance measurement exists yet.

Amazonverified