Amazon unveils TorchNeuron to make its Trainium AI chips easier to use

Amazon wants developers to stop thinking of its AI chips as a science project. On October 19, 2026, the company unveiled TorchNeuron, an open-source tool that lets PyTorch code run on AWS Trainium chips with a single line changed. Wall Street noticed. Amazon shares climbed as much as 5% intraday after the announcement. What TorchNeuron actually does TorchNeuron is a PyTorch backend built by Annapurna Labs, the Amazon division responsible for its custom chip designs. It registers Trainium as a native device inside PyTorch, the widely used framework for building and training AI models. In practical terms, developers can take PyTorch code they already wrote and point it at Trainium2 or Trainium3 instances. The switch is a one-line device change, rather than a lengthy rewrite. The public beta opened on October 20, 2026, during the PyTorch Conference in San Jose. Amazon released it under the Apache 2.0 license, a permissive open-source license that allows broad commercial use. The tool supports the core PyTorch features developers expect. That includes eager execution, distributed training APIs, and the Neuron Kernel Interface for writing custom kernels. The demo and the early adopters ...