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. Amazon has launched TorchNeuron, a tool that lets developers […] This story continues at The Next Web

Read it at The Next Web ↗

← Front page