diff --git a/docs/advanced_guide.rst b/docs/advanced_guide.rst index d0d51cf2e..5e78faf07 100644 --- a/docs/advanced_guide.rst +++ b/docs/advanced_guide.rst @@ -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 \ No newline at end of file diff --git a/docs/index.rst b/docs/index.rst index 5157a0b59..7d2984edb 100644 --- a/docs/index.rst +++ b/docs/index.rst @@ -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 \ No newline at end of file diff --git a/docs/notes.rst b/docs/notes.rst index 10a5d95b1..025067fd3 100644 --- a/docs/notes.rst +++ b/docs/notes.rst @@ -12,4 +12,5 @@ Notes gpu_memory_allocation profiling device_memory_profiling - rank_promotion_warning \ No newline at end of file + rank_promotion_warning + jax_array_migration \ No newline at end of file diff --git a/docs/user_guides.rst b/docs/user_guides.rst index 9e2c5aaf9..b78f0609f 100644 --- a/docs/user_guides.rst +++ b/docs/user_guides.rst @@ -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 \ No newline at end of file