This document is relevant for: Inf1, Inf2, Trn1, Trn2

Neuron GitHub Samples

Training Samples for Trn1

• PyTorch Neuron (torch-neuronx) samples for Trn1

• Nemo Megatron for Neuron for Trn1

• AWS Neuron samples for ParallelCluster

• AWS Neuron samples for EKS

• AWS Neuron samples for SageMaker

• AWS Neuron samples for Batch

Inference Samples for Inf2 & Trn1

• PyTorch Neuron (torch-neuronx) samples for Inf2 & Trn1

• Transformers Neuron (transformers-neuronx) samples

• AWS Neuron samples for SageMaker

Inference Samples for Inf1

• PyTorch Neuron (torch-neuron) samples for Inf1

• TensorFlow Neuron (tensorflow-neuron) samples for Inf1

This document is relevant for: Inf1, Inf2, Trn1, Trn2