.. _jax-neuron-main:

JAX Neuron (beta)
=================

JAX Neuron 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.


.. toctree::
        :maxdepth: 1
        :hidden:

        /frameworks/jax/setup/jax-setup
        /frameworks/jax/setup/jax-neuronx-known-issues
        /frameworks/jax/api-reference-guide/index
        /release-notes/jax-neuronx/index.rst

* :ref:`jax-neuron-setup`
* :ref:`jax-neuron-known-issues`
* :ref:`jax-neuronx-api-reference-guide`
* :ref:`jax-neuronx-rn`