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
Run training in Pytorch Neuron container
Deploy a simple mlp training script as a Kubernetes job
Run inference in pytorch neuron container
Deploy a TensorFlow Resnet50 model as a Kubernetes service