.. _jax-tutorials:

Tutorials
=========

.. toctree::
   :maxdepth: 1

   quickstart
   key-concepts
   jit-compilation
   automatic-vectorization
   automatic-differentiation
   debugging
   random-numbers
   working-with-pytrees
   sharded-computation
   stateful-computations
   control-flow
   
.. toctree::
   :maxdepth: 1
   :caption: Advanced tutorials

   advanced-autodiff
   external-callbacks
   gradient-checkpointing
   jax-primitives
   jaxpr