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
Inf1
Running Inference on Variable Input Shapes with Bucketing
Data Parallel Inference on PyTorch Neuron
Developer Guide - PyTorch Neuron (torch-neuron) LSTM Support
PyTorch Neuron (torch-neuron) Core Placement