This document is relevant for: Inf2, Trn1, Trn1n

API Reference Guide (torch-neuronx)

API Reference Guide (torch-neuronx)

• PyTorch Neuron (torch-neuronx) Tracing API for Inference

• PyTorch Neuron (torch-neuronx) NeuronCore Placement APIs [Experimental]

• PyTorch Neuron (torch-neuronx) Analyze API for Inference

This document is relevant for: Inf2, Trn1, Trn1n