This document is relevant for: Trn1, Trn1n

API Reference Guide for Training (torch-neuronx)

API Reference Guide for Training (torch-neuronx)

• PyTorch Neuron neuron_parallel_compile CLI (torch-neuronx)

• PyTorch Neuron Environment Variables (torch-neuronx)

• PyTorch Neuron (torch-neuronx) Profiling API

This document is relevant for: Trn1, Trn1n