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 enters public beta on October 20 under the Apache 2.0 license, supports Trainium2 and Trainium3, and does not yet cover every PyTorch feature.