This document is relevant for: Trn2, Trn3

Native PyTorch for AWS Trainium

Important

This page has moved. The TorchNeuron native PyTorch documentation is now at TorchNeuron: Native PyTorch on AWS Neuron.

• TorchNeuron overview — Architecture, eager mode, torch.compile, distributed training, NKI

• TorchTitan on Neuron — Pre-training with TorchTitan on TorchNeuron

• Get started — Quickstart guides for eager mode and torch.compile

Please update your bookmarks.

This document is relevant for: Trn2, Trn3