:orphan:

.. _jax-tutorials:

JAX tutorials
=============

.. note::

   The tutorials below are a work in progress; for the time being, please refer
   to the older tutorial content at :ref:`Jax-101`, :ref:`beginner-guide` and
   :ref:`user-guides`.

JAX 101
-------

.. toctree::
   :maxdepth: 1

   installation
   quickstart
   jax-as-accelerated-numpy
   jit-compilation
   automatic-vectorization
   automatic-differentiation
   random-numbers
   working-with-pytrees
   single-host-sharding
   stateful-computations
   simple-neural-network


JAX 201
-------

.. toctree::
   :maxdepth: 1

   parallelism
   advanced-autodiff
   external-callbacks
   debugging
   profiling-and-performance


JAX 301
-------

.. toctree::
   :maxdepth: 1

   jax-internals
   advanced-compilation