Overview
ML Frameworks
torch-neuronx
torch-neuron
LSTM
tensorflow-neuronx
tensorflow-neuron
tensorflow-neuron (TF1.x)
tensorflow-neuron (TF2.x)
ML Libraries
transformers-neuronx
neuronx-distributed
User Guide
neuronx-cc
neuron-cc
Learning Neuron
About Neuron
This document is relevant for: Inf1, Inf2, Trn1, Trn1n
Inf1
Inf2
Trn1
Trn1n
Bring Your Own Neuron Container to Sagemaker Hosting (inf2 or trn1)
Bring Your Own Neuron Container to Sagemaker Hosting (inf1)
Compile with Sagemaker Neo and Deploy on Sagemaker Hosting (inf1)
AWS Neuron Sagemaker Samples GitHub Repository
Train your model on SageMaker