mirror of
https://github.com/ROCm/jax.git
synced 2025-04-17 20:36:05 +00:00
25 lines
355 B
ReStructuredText
25 lines
355 B
ReStructuredText
:orphan:
|
|
|
|
.. _Jax-101:
|
|
|
|
Tutorial: JAX 101
|
|
=================
|
|
|
|
This is a tutorial developed by engineers and researchers at DeepMind_.
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
:caption: Tutorials
|
|
|
|
01-jax-basics
|
|
02-jitting
|
|
03-vectorization
|
|
04-advanced-autodiff
|
|
05-random-numbers
|
|
05.1-pytrees
|
|
06-parallelism
|
|
07-state
|
|
|
|
|
|
.. _Deepmind: http://deepmind.com
|