This document is relevant for: Inf2, Trn1

TensorFlow NeuronX (Inf2 & Trn1) — Archived

Warning

This document is archived. TensorFlow NeuronX (tensorflow-neuronx) is no longer officially supported by the AWS Neuron SDK. It is provided for reference only. For current framework support, see ML framework support on AWS Neuron SDK.

TensorFlow NeuronX (tensorflow-neuronx) was the TensorFlow integration for AWS Inferentia2 (Inf2) and Trainium (Trn1) instances. These pages are preserved for historical reference alongside the Inference (Inf2 & Trn1) landing.

This document is relevant for: Inf2, Trn1