mirror of
https://github.com/ROCm/jax.git
synced 2025-04-25 17:46:05 +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
|