mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Consolidate links in JAX documentation

Browse Source 
Move notes down
This commit is contained in:
Ravin Kumar 2022-11-08 14:45:58 -08:00
parent b07c586565
commit cb3762e5dd
1 changed files with 18 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
docs/index.rst
View File

@ -15,7 +15,7 @@ parallelize, Just-In-Time compile to GPU/TPU, and more.
notebooks/Common_Gotchas_in_JAX
.. toctree::
:maxdepth: 2
:maxdepth: 1
jax-101/index
@ -54,6 +54,23 @@ parallelize, Just-In-Time compile to GPU/TPU, and more.
notebooks/xmap_tutorial
multi_process
.. toctree::
:maxdepth: 1
:caption: Developer documentation
contributing
developer
jax_internal_api
autodidax
jep/index
.. toctree::
:maxdepth: 1
:caption: API documentation
jax
.. toctree::
:maxdepth: 1
:caption: Notes
@ -66,22 +83,6 @@ parallelize, Just-In-Time compile to GPU/TPU, and more.
device_memory_profiling
rank_promotion_warning
.. toctree::
:maxdepth: 2
:caption: Developer documentation
contributing
developer
jax_internal_api
autodidax
jep/index
.. toctree::
:maxdepth: 3
:caption: API documentation
jax
Indices and tables
==================