JAX Framework
=================

.. toctree::
   :maxdepth: 2

   jax-getting_started.ipynb
   jax-basic_example.ipynb
   jax-multigpu_example.ipynb
   flax-basic_example.ipynb
   pax-basic_example.ipynb
   t5x-basic_example.ipynb