This document is relevant for: Trn2, Trn3

TorchNeuron API reference

API reference material for torch-neuronx on AWS Neuron.

Operator compatibility table

Per-operator support status, dtype coverage, and CPU fallbacks for torch-neuronx, maintained in the torch-neuronx repository.

torch.compile backend reference

torch.compile parameter and torch.compiler API support, graph breaks, environment variables, and caching.

This document is relevant for: Trn2, Trn3