Comment Comment by npalli Interesting, Amazon doing the same for AWS Trainium.https://github.com/aws-neuron/torch-neuronx parent npalli 11 days ago
https://github.com/aws-neuron/torch-neuronx