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 ../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