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

Pytorch Neuron Setup

PyTorch Neuron (torch-neuronx) Setup for Inf2, Trn1, and Trn2 Instances

PyTorch Neuron (torch-neuron) Setup for Inf1 Instances

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