mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 11:56:07 +00:00
Reorganization among notes, user guides, advanced guides
This commit is contained in:
parent
3df9250169
commit
e5f0103895
@ -16,5 +16,6 @@ Advanced Tutorials
|
||||
notebooks/Writing_custom_interpreters_in_Jax
|
||||
notebooks/Neural_Network_and_Data_Loading
|
||||
notebooks/xmap_tutorial
|
||||
notebooks/convolutions
|
||||
notebooks/external_callbacks
|
||||
Custom_Operation_for_GPUs
|
@ -111,6 +111,14 @@ installation options, see the `Install Guide`_ in the project README.
|
||||
jax
|
||||
|
||||
|
||||
.. toctree::
|
||||
:hidden:
|
||||
:maxdepth: 1
|
||||
|
||||
changelog
|
||||
glossary
|
||||
|
||||
|
||||
.. _Autograd: https://github.com/hips/autograd
|
||||
.. _XLA: https://www.tensorflow.org/xla
|
||||
.. _Install Guide: https://github.com/google/jax#installation
|
@ -12,4 +12,5 @@ Notes
|
||||
gpu_memory_allocation
|
||||
profiling
|
||||
device_memory_profiling
|
||||
rank_promotion_warning
|
||||
rank_promotion_warning
|
||||
jax_array_migration
|
@ -10,12 +10,8 @@ User Guides
|
||||
async_dispatch
|
||||
aot
|
||||
jaxpr
|
||||
notebooks/convolutions
|
||||
pytrees
|
||||
jax_array_migration
|
||||
type_promotion
|
||||
errors
|
||||
transfer_guard
|
||||
glossary
|
||||
changelog
|
||||
debugging/index
|
Loading…
x
Reference in New Issue
Block a user