mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 11:56:07 +00:00
30 lines
457 B
ReStructuredText
30 lines
457 B
ReStructuredText
.. _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
|