Tutorial: JAX 101
=================

This is a tutorial developed by engineers and researchers at DeepMind_.

.. toctree::
  :maxdepth: 1
  :caption: Tutorials

  01-jax-basics
  02-jitting
  03-vectorization
  04-advanced-autodiff
  05-random-numbers
  05.1-pytrees
  06-parallelism
  07-state
  08-pjit


.. _Deepmind: http://deepmind.com