:orphan:

.. _jax-tutorials-draft:

JAX tutorials draft
===================

.. note::

   The tutorials below are a work in progress; for the time being, please refer
   to the older tutorial content, including :ref:`beginner-guide`,
   :ref:`jax-101`, and the now-deleted *JAX 101* tutorials.

JAX 101
-------
Mostly finalized at :ref:`jax-101`!

.. toctree::
   :maxdepth: 1

   ../key-concepts
   ../jit-compilation
   ../automatic-vectorization
   ../automatic-differentiation
   ../debugging
   ../random-numbers
   ../pytrees
   ../parallel
   ../stateful-computations
   simple-neural-network


JAX 201
-------

.. toctree::
   :maxdepth: 1

   parallelism
   advanced-debugging
   profiling-and-performance

JAX 301
-------

.. toctree::
   :maxdepth: 1

   advanced-compilation
