TorchNeuron

TorchNeuron enables PyTorch machine learning code to execute directly on Amazon AWS Trainium hardware. It allows developers to migrate model workloads without rewriting underlying code bases.