diff --git a/frameworks/jax/index.rst b/frameworks/jax/index.rst index 5b1a6f1e..87feb8ad 100644 --- a/frameworks/jax/index.rst +++ b/frameworks/jax/index.rst @@ -3,7 +3,7 @@ JAX Neuron (beta) ================= -JAX Neuron is a software package containing Neuron-specific JAX features, such as +JAX NeuronX is a software package containing Neuron-specific JAX features, such as the Neuron NKI JAX interface. It also serves as a meta-package for providing a tested combination of the ``jax-neuronx``, ``jax``, ``jaxlib``, ``libneuronxla`` , and ``neuronx-cc`` packages.