API Reference Guide for Training (torch-neuronx) — AWS Neuron Documentation (original) (raw)

• Repository

• Suggest edit

• Open issue

• .rst

This document is relevant for: Trn1, Trn2

API Reference Guide for Training (torch-neuronx)#

API Reference Guide for Training (torch-neuronx)

• PyTorch NeuronX neuron_parallel_compile CLI
• Neuron Persistent Cache
• PyTorch NeuronX Environment Variables
• PyTorch NeuronX Profiling API

This document is relevant for: Trn1, Trn2