Introducing TorchNeuron: Native PyTorch backend for AWS Trainium (27 minute read)
PyTorch announced TorchNeuron, an open-source native PyTorch backend for AWS Trainium, with public beta availability scheduled for October 20 under the Apache 2.0 license.
source-linked1