.. _training-torch-neuronx:


.. meta::
   :description: Training (``torch-neuronx``) - AWS Neuron SDK documentation
   :keywords: AWS Neuron, Inferentia, PyTorch, Trainium, torch-neuronx, training
   :date-modified: 2026-03-13


Training (``torch-neuronx``)
============================

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

.. toctree::
    :maxdepth: 1
    :hidden:

     Tutorials </frameworks/torch/torch-neuronx/tutorials/training/tutorials-training-torch-neuronx>
     Additional Examples </frameworks/torch/torch-neuronx/additional-examples-training>
     API Reference Guide </frameworks/torch/torch-neuronx/api-reference-guide/training/index>
     Developer Guide  </frameworks/torch/torch-neuronx/programming-guide/training/index>
     Misc  </frameworks/torch/torch-neuronx/misc-training>

Get Started
------------

.. grid:: 2
    :gutter: 2

    .. grid-item-card:: Setup (``torch-neuronx``)
        :link: setup-torch-neuronx
        :link-type: ref
        :class-header: sd-bg-primary sd-text-white

        Install and configure PyTorch NeuronX for training workloads.

Tutorials & Examples
---------------------

.. grid:: 2
    :gutter: 2

    .. grid-item-card:: Tutorials
        :link: /frameworks/torch/torch-neuronx/tutorials/training/tutorials-training-torch-neuronx
        :link-type: doc
        :class-header: sd-bg-primary sd-text-white

        Step-by-step training tutorials for PyTorch NeuronX.

    .. grid-item-card:: Additional Examples
        :link: /frameworks/torch/torch-neuronx/additional-examples-training
        :link-type: doc
        :class-header: sd-bg-primary sd-text-white

        More training examples and sample code.

Reference
----------

.. grid:: 2
    :gutter: 2

    .. grid-item-card:: API Reference Guide
        :link: /frameworks/torch/torch-neuronx/api-reference-guide/training/index
        :link-type: doc
        :class-header: sd-bg-primary sd-text-white

        Training API reference for PyTorch NeuronX.

    .. grid-item-card:: Developer Guide
        :link: /frameworks/torch/torch-neuronx/programming-guide/training/index
        :link-type: doc
        :class-header: sd-bg-primary sd-text-white

        In-depth developer guide for training on Neuron.

    .. grid-item-card:: Misc
        :link: /frameworks/torch/torch-neuronx/misc-training
        :link-type: doc
        :class-header: sd-bg-primary sd-text-white

        Supported operators, multi-node setup, and troubleshooting.