
.. meta::
   :description: API Reference Guide  (``torch-neuronx``) - AWS Neuron SDK documentation
   :keywords: API reference, AWS Neuron, Inferentia, PyTorch, Trainium, inference, torch-neuronx
   :date-modified: 2026-03-13


API Reference Guide  (``torch-neuronx``)
========================================

.. toctree::
    :maxdepth: 1
    :hidden:
    
    /frameworks/torch/torch-neuronx/api-reference-guide/inference/api-torch-neuronx-trace
    /frameworks/torch/torch-neuronx/api-reference-guide/inference/api-torch-neuronx-replace-weights
    /frameworks/torch/torch-neuronx/api-reference-guide/inference/api-torch-neuronx-core-placement
    /frameworks/torch/torch-neuronx/api-reference-guide/inference/api-torch-neuronx-analyze
    /frameworks/torch/torch-neuronx/api-reference-guide/inference/api-torch-neuronx-data-parallel


.. dropdown::  API Reference Guide  (``torch-neuronx``)
    :class-title: sphinx-design-class-title-small
    :class-body: sphinx-design-class-body-small
    :animate: fade-in
    :open:

    * :ref:`torch_neuronx_trace_api`
    * :ref:`torch_neuronx_replace_weights_api`
    * :ref:`torch_neuronx_core_placement_api`
    * :ref:`torch_neuronx_analyze_api`
    * :ref:`torch_neuronx_dataparallel_api`
    * :ref:`torch_neuronx_lazy_async_load_api`

