This document is relevant for: Inf2, Trn1, Trn2

Training (`torch-neuronx`)#

Train models using PyTorch NeuronX on Trainium instances (Trn1, Trn2).

Get Started#

Setup (torch-neuronx)

Install and configure PyTorch NeuronX for training workloads.

Tutorials

Step-by-step training tutorials for PyTorch NeuronX.

Additional Examples

More training examples and sample code.

API Reference Guide

Training API reference for PyTorch NeuronX.

Developer Guide

In-depth developer guide for training on Neuron.

Misc

Supported operators, multi-node setup, and troubleshooting.

This document is relevant for: Inf2, Trn1, Trn2