TorchNeuron lets PyTorch code run on AWS Trainium unmodified
AWS and the PyTorch team ship an open-source native backend for Trainium chips, in public beta from 20 October. Existing PyTorch code switches device with one line.
Tagged
Tagged
AWS and the PyTorch team ship an open-source native backend for Trainium chips, in public beta from 20 October. Existing PyTorch code switches device with one line.