https://github.com/google/jax
Raw File
Tip revision: 9e62994bce7c7fcbb2f6a50c9ef89526cd2c2be6 authored by jax authors on 09 May 2024, 19:32:51 UTC
Merge pull request #21135 from hawkinsp:release
Tip revision: 9e62994
tutorials.rst
.. _jax-tutorials:

JAX tutorials
=============

.. toctree::
   :maxdepth: 1

   quickstart
   key-concepts
   jit-compilation
   automatic-vectorization
   automatic-differentiation
   debugging
   random-numbers
   working-with-pytrees
   sharded-computation
   stateful-computations
back to top