diff --git a/docs/advanced_guide.rst b/docs/advanced_guide.rst
index 5e78faf07..305b5d8b9 100644
--- a/docs/advanced_guide.rst
+++ b/docs/advanced_guide.rst
@@ -2,20 +2,41 @@
 
 Advanced Tutorials
 ==================
+This section contains examples and tutorials on more advanced topics, such as Multi Core computation, Custom operations, and more in depth applications
 
 .. toctree::
+   :caption: Examples
+   :maxdepth: 1
+
+   notebooks/neural_network_with_tfds_data
+   notebooks/Neural_Network_and_Data_Loading
+   notebooks/vmapped_log_probs
+
+.. toctree::
+   :caption: Parallel Computation
+   :maxdepth: 1
+
+   multi_process
+   notebooks/Distributed_arrays_and_automatic_parallelization
+   notebooks/xmap_tutorial
+
+.. toctree::
+   :caption: Automatic Differentiation
    :maxdepth: 1
 
    notebooks/autodiff_cookbook
-   multi_process
-   notebooks/Distributed_arrays_and_automatic_parallelization
-   notebooks/vmapped_log_probs
-   notebooks/neural_network_with_tfds_data
    notebooks/Custom_derivative_rules_for_Python_code
+
+.. toctree::
+   :caption: JAX Internals
+   :maxdepth: 1
+
    notebooks/How_JAX_primitives_work
    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
+   Custom_Operation_for_GPUs
+
+.. toctree::
+   :caption: Deep Dives
+   :maxdepth: 1
+
+   notebooks/convolutions
\ No newline at end of file
diff --git a/docs/notebooks/convolutions.ipynb b/docs/notebooks/convolutions.ipynb
index fa833866c..2cb07d763 100644
--- a/docs/notebooks/convolutions.ipynb
+++ b/docs/notebooks/convolutions.ipynb
@@ -6,7 +6,7 @@
     "id": "TVT_MVvc02AA"
    },
    "source": [
-    "# Convolutions in JAX\n",
+    "# Generalized Convolutions in JAX\n",
     "\n",
     "[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/jax/blob/main/docs/notebooks/convolutions.ipynb)\n",
     "\n",
diff --git a/docs/notebooks/convolutions.md b/docs/notebooks/convolutions.md
index cffe73da9..261c1b181 100644
--- a/docs/notebooks/convolutions.md
+++ b/docs/notebooks/convolutions.md
@@ -14,7 +14,7 @@ kernelspec:
 
 +++ {"id": "TVT_MVvc02AA"}
 
-# Convolutions in JAX
+# Generalized Convolutions in JAX
 
 [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/jax/blob/main/docs/notebooks/convolutions.ipynb)
 
diff --git a/docs/notebooks/vmapped_log_probs.ipynb b/docs/notebooks/vmapped_log_probs.ipynb
index 75f731ab0..c450e6487 100644
--- a/docs/notebooks/vmapped_log_probs.ipynb
+++ b/docs/notebooks/vmapped_log_probs.ipynb
@@ -6,7 +6,7 @@
     "id": "6umP1IKf4Dg6"
    },
    "source": [
-    "# Autobatching log-densities example\n",
+    "# Autobatching for Bayesian Inference\n",
     "\n",
     "[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/jax/blob/main/docs/notebooks/vmapped_log_probs.ipynb)\n",
     "\n",
diff --git a/docs/notebooks/vmapped_log_probs.md b/docs/notebooks/vmapped_log_probs.md
index 890907cfc..c96d36427 100644
--- a/docs/notebooks/vmapped_log_probs.md
+++ b/docs/notebooks/vmapped_log_probs.md
@@ -14,7 +14,7 @@ kernelspec:
 
 +++ {"id": "6umP1IKf4Dg6"}
 
-# Autobatching log-densities example
+# Autobatching for Bayesian Inference
 
 [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/jax/blob/main/docs/notebooks/vmapped_log_probs.ipynb)
 
diff --git a/docs/notes.rst b/docs/notes.rst
index 940b0373e..aead9e23e 100644
--- a/docs/notes.rst
+++ b/docs/notes.rst
@@ -11,4 +11,5 @@ Notes
    concurrency
    gpu_memory_allocation
    rank_promotion_warning
-   jax_array_migration
\ No newline at end of file
+   jax_array_migration
+   type_promotion
\ No newline at end of file
diff --git a/docs/user_guides.rst b/docs/user_guides.rst
index a96882ea9..dc5825614 100644
--- a/docs/user_guides.rst
+++ b/docs/user_guides.rst
@@ -11,9 +11,9 @@ User Guides
    aot
    jaxpr
    pytrees
-   type_promotion
    errors
    debugging/index
    profiling
    device_memory_profiling
    transfer_guard
+   notebooks/external_callbacks