This document is relevant for: Inf1, Inf2, Trn1, Trn1n

Pytorch Neuron Setup

PyTorch Neuron (torch-neuronx) Setup for Inf2, Trn1/Trn1n Instances

PyTorch Neuron (torch-neuron) Setup for Inf1 Instances

This document is relevant for: Inf1, Inf2, Trn1, Trn1n