rocm_jax/docs/tutorials.rst

.. _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
DOC: replace old tutorials with new content 2024-04-18 13:11:25 -07:00			`.. _jax-tutorials:`

docs: tidy up titles and headings This shortens some titles and makes them more consistent. It also removes "JAX" from several titles ("in JAX", "for JAX", "JAX's", etc.). Since these are JAX docs, that ought to be clear from context. 2024-08-12 22:13:36 -07:00			`Tutorials`
			`=========`
DOC: replace old tutorials with new content 2024-04-18 13:11:25 -07:00
			`.. toctree::`
			`:maxdepth: 1`

			`quickstart`
			`key-concepts`
			`jit-compilation`
			`automatic-vectorization`
			`automatic-differentiation`
			`debugging`
			`random-numbers`
			`working-with-pytrees`
			`sharded-computation`
			`stateful-computations`
Move Control Flow text from Sharp Bits into its own tutorial. 2024-11-06 10:43:17 -08:00			`control-flow`
Add/update JAX Advanced Tutorials docs, ToC structure 2024-09-20 22:19:14 +00:00
			`.. toctree::`
			`:maxdepth: 1`
			`:caption: Advanced tutorials`

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