.. _jax-101:

JAX 101

These tutorials cover basic usage of JAX and its features, including some of the internal mechanisms that make JAX work. They start with the fundamentals and are meant to be read sequentially. For more in-depth discussions of JAX's design and implementation, in no particular order, see the :doc:advanced guides <advanced_guides>.

.. toctree:: :maxdepth: 1

jit-compilation automatic-vectorization automatic-differentiation pytrees random-numbers sharded-computation control-flow tracing stateful-computations