Amazon’s TorchNeuron lets PyTorch code run on AWS Trainium chips
Amazon has launched TorchNeuron, an open-source backend that lets PyTorch code run on its Trainium AI chips with a one-line change. It goes into public beta on 20 October under the Apache 2.0 licence and supports Trainium2 and Trainium3. Some PyTorch features are not yet supported.
The Next Web AI2 min read