Announcing Name Change for Neuron Components — AWS Neuron Documentation (original) (raw)

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

Announcing Name Change for Neuron Components#

Starting with Neuron release 2.16, the name of the following Neuron components will change as follows:

Package name Current Name New Name
torch-neuronx PyTorch Neuron PyTorch NeuronX
tensorflow-neuronx TensorFlow Neuron TensorFlow NeuronX
neuronx-cc Neuron Compiler NeuronX Compiler
aws-neuronx-runtime-lib Neuron Runtime NeuronX Runtime
tensorflow-neuronx Transformers Neuron Transformers NeuronX
neuronx-distributed Neuron Distributed NeuronX Distributed

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