Amazon's TorchNeuron lets PyTorch code run on AWS Trainium chips
Amazon · Products
Amazon says TorchNeuron lets developers run their existing PyTorch code on AWS Trainium chips with a single configuration change and no rewrite. In a side-by-side test, Amazon engineers ran the same AI training script on a conventional chip and on Trainium, and the model learned the same things at the same rate. TorchNeuron supports Trainium2 and Trainium3 and will be released under the Apache 2.0 open-source license at the PyTorch Conference North America on October 20.