Amazon launches TorchNeuron to bring PyTorch code to Trainium chips

First reported by Hugging Face at · Updated · 5 sources

The open-source software lets developers use existing PyTorch code on Amazon’s AI hardware with only a basic configuration change. Amazon is making the tool available in public beta, offering developers another option for running their AI applications without substantial code changes when moving to a different chip platform.

  • According to The Next Web, TorchNeuron supports Trainium2 and Trainium3.
  • The Next Web reports an Apache 2.0 licence and some unsupported PyTorch features.

Covered by 5 publishers within 20 hours of the first report: 3 news reports and 2 from the company.

Reporting3

GeekWire Amazon challenges Nvidia’s software advantage with new tool for its own AI chips · Todd Bishop
The Next Web Amazon’s TorchNeuron lets PyTorch code run on AWS Trainium chips · Ana Maria Constantin
TechCrunch Amazon is taking its biggest shot at Nvidia yet with new TorchNeuron dev tool · Julie Bort

From the company2

Hugging Face TorchNeuron: running HF models on AWS Trainium is easier than ever
Amazon Amazon opens its AWS Trainium AI chips to every PyTorch developer, no rewrite needed

Related

Amazon S3 is now available on the Vercel Marketplace and v0 by Vercel
Pinterest appoints Amazon executive James Dibbo as CFO
Amazon Reportedly in Talks to Buy Decart AI for Around $7B
Amazon's new Alexa tablets ditch Fire OS for regular Android

Topics Amazon