Amazon’s TorchNeuron lets PyTorch code run on AWS Trainium chips

T2Covered by 1 source · first reported by TNW at 19 Oct, 16:21 UTC

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. Amazon has…