mirror of
https://github.com/ROCm/jax.git
synced 2025-04-27 02:06:06 +00:00
54 lines
787 B
ReStructuredText
54 lines
787 B
ReStructuredText
:orphan:
|
|
|
|
.. _jax-tutorials:
|
|
|
|
JAX tutorials
|
|
=============
|
|
|
|
.. note::
|
|
|
|
The tutorials below are a work in progress; for the time being, please refer
|
|
to the older tutorial content at :ref:`Jax-101`, :ref:`beginner-guide` and
|
|
:ref:`user-guides`.
|
|
|
|
JAX 101
|
|
-------
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
|
|
installation
|
|
quickstart
|
|
jax-as-accelerated-numpy
|
|
jit-compilation
|
|
automatic-vectorization
|
|
automatic-differentiation
|
|
random-numbers
|
|
working-with-pytrees
|
|
single-host-sharding
|
|
stateful-computations
|
|
simple-neural-network
|
|
|
|
|
|
JAX 201
|
|
-------
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
|
|
parallelism
|
|
advanced-autodiff
|
|
external-callbacks
|
|
debugging
|
|
profiling-and-performance
|
|
|
|
|
|
JAX 301
|
|
-------
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
|
|
jax-internals
|
|
advanced-compilation
|