mirror of
https://github.com/ROCm/jax.git
synced 2025-04-25 08:26:07 +00:00
58 lines
950 B
ReStructuredText
58 lines
950 B
ReStructuredText
:orphan:
|
|
|
|
.. _jax-tutorials-draft:
|
|
|
|
JAX tutorials draft
|
|
===================
|
|
|
|
.. note::
|
|
|
|
This is a
|
|
The tutorials below are a work in progress; for the time being, please refer
|
|
to the older tutorial content, including :ref:`beginner-guide`,
|
|
:ref:`user-guides`, and the now-deleted *JAX 101* tutorials.
|
|
|
|
JAX 101
|
|
-------
|
|
Mostly finalized at :ref:`jax-tutorials`!
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
|
|
../quickstart
|
|
../key-concepts
|
|
../jit-compilation
|
|
../automatic-vectorization
|
|
../automatic-differentiation
|
|
../debugging
|
|
../random-numbers
|
|
../working-with-pytrees
|
|
../sharded-computation
|
|
../stateful-computations
|
|
simple-neural-network
|
|
|
|
|
|
JAX 201
|
|
-------
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
|
|
parallelism
|
|
advanced-autodiff
|
|
gradient-checkpointing
|
|
advanced-debugging
|
|
external-callbacks
|
|
profiling-and-performance
|
|
|
|
|
|
JAX 301
|
|
-------
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
|
|
jax-primitives
|
|
jaxpr
|
|
advanced-compilation
|