This document is relevant for: Trn1, Trn2
Tutorials (torch-neuronx)#
- Hugging Face BERT Pretraining Tutorial (Data-Parallel)
- Multi-Layer Perceptron Training Tutorial
- PyTorch Neuron for Trainium Hugging Face BERT MRPC task finetuning using Hugging Face Trainer API
- ZeRO-1 Tutorial
- Analyze for Training Tutorial
- Neuron Custom C++ Operators in MLP Training
- Neuron Custom C++ Operators Performance Optimization
This document is relevant for: Trn1, Trn2