index.rst

orphan:

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:`user-guides`, and the now-deleted JAX 101 tutorials.

JAX 101

Mostly finalized at :ref:`jax-tutorials`!

.. toctree::
   :maxdepth: 1

   ../quickstart
   ../key-concepts
   ../jit-compilation
   ../automatic-vectorization
   ../automatic-differentiation
   ../debugging
   ../random-numbers
   ../working-with-pytrees
   ../sharded-computation
   ../stateful-computations
   simple-neural-network

JAX 201

.. toctree::
   :maxdepth: 1

   parallelism
   advanced-debugging
   profiling-and-performance

JAX 301

.. toctree::
   :maxdepth: 1

   advanced-compilation