garymm
diff --git a/‎docs/Custom_Operation_for_GPUs.md
+2 b/‎docs/Custom_Operation_for_GPUs.md
+2
diff --git a/‎docs/_tutorials/advanced-autodiff.md
+2 b/‎docs/_tutorials/advanced-autodiff.md
+2
diff --git a/‎docs/_tutorials/advanced-compilation.md
+2 b/‎docs/_tutorials/advanced-compilation.md
+2
diff --git a/‎docs/_tutorials/advanced-debugging.md
+3 b/‎docs/_tutorials/advanced-debugging.md
+3
diff --git a/‎docs/_tutorials/external-callbacks.md
+2 b/‎docs/_tutorials/external-callbacks.md
+2
diff --git a/‎docs/_tutorials/gradient-checkpointing.md
+2 b/‎docs/_tutorials/gradient-checkpointing.md
+2
diff --git a/‎docs/_tutorials/jax-primitives.md
+2 b/‎docs/_tutorials/jax-primitives.md
+2
diff --git a/‎docs/_tutorials/jaxpr.md
+2 b/‎docs/_tutorials/jaxpr.md
+2
diff --git a/‎docs/_tutorials/parallelism.md
+2 b/‎docs/_tutorials/parallelism.md
+2
diff --git a/‎docs/_tutorials/profiling-and-performance.md
+2 b/‎docs/_tutorials/profiling-and-performance.md
+2
diff --git a/‎docs/_tutorials/simple-neural-network.md
+2 b/‎docs/_tutorials/simple-neural-network.md
+2
diff --git a/‎docs/aot.md
+2 b/‎docs/aot.md
+2
diff --git a/‎docs/api_compatibility.md
+2 b/‎docs/api_compatibility.md
+2
diff --git a/‎docs/autodidax.ipynb
+2 b/‎docs/autodidax.ipynb
+2
diff --git a/‎docs/autodidax.md
+2 b/‎docs/autodidax.md
+2
diff --git a/‎docs/autodidax.py
+2 b/‎docs/autodidax.py
+2
diff --git a/‎docs/automatic-differentiation.md
+2 b/‎docs/automatic-differentiation.md
+2
diff --git a/‎docs/automatic-vectorization.md
+2 b/‎docs/automatic-vectorization.md
+2
diff --git a/‎docs/building_on_jax.md
+2 b/‎docs/building_on_jax.md
+2
diff --git a/‎docs/contributing.md
+2 b/‎docs/contributing.md
+2
diff --git a/‎docs/debugging.md
+2 b/‎docs/debugging.md
+2
diff --git a/‎docs/debugging/checkify_guide.md
+2 b/‎docs/debugging/checkify_guide.md
+2
diff --git a/‎docs/debugging/flags.md
+2 b/‎docs/debugging/flags.md
+2
diff --git a/‎docs/debugging/index.md
+2 b/‎docs/debugging/index.md
+2
diff --git a/‎docs/debugging/print_breakpoint.md
+2 b/‎docs/debugging/print_breakpoint.md
+2
diff --git a/‎docs/deprecation.md
+2 b/‎docs/deprecation.md
+2
diff --git a/‎docs/developer.md
+2 b/‎docs/developer.md
+2
diff --git a/‎docs/device_memory_profiling.md
+1 b/‎docs/device_memory_profiling.md
+1
diff --git a/‎docs/distributed_data_loading.md
+2 b/‎docs/distributed_data_loading.md
+2
diff --git a/‎docs/gpu_performance_tips.md
+2 b/‎docs/gpu_performance_tips.md
+2
diff --git a/‎docs/installation.md
+2 b/‎docs/installation.md
+2
diff --git a/‎docs/investigating_a_regression.md
+2 b/‎docs/investigating_a_regression.md
+2
diff --git a/‎docs/jax_array_migration.md
+2 b/‎docs/jax_array_migration.md
+2
diff --git a/‎docs/jit-compilation.md
+2 b/‎docs/jit-compilation.md
+2
diff --git a/‎docs/key-concepts.md
+2 b/‎docs/key-concepts.md
+2
diff --git a/‎docs/multi_process.md
+2 b/‎docs/multi_process.md
+2
diff --git a/‎docs/notebooks/Common_Gotchas_in_JAX.ipynb
+2 b/‎docs/notebooks/Common_Gotchas_in_JAX.ipynb
+2
diff --git a/‎docs/notebooks/Common_Gotchas_in_JAX.md
+2 b/‎docs/notebooks/Common_Gotchas_in_JAX.md
+2
diff --git a/‎docs/notebooks/Custom_derivative_rules_for_Python_code.ipynb
+2 b/‎docs/notebooks/Custom_derivative_rules_for_Python_code.ipynb
+2
diff --git a/‎docs/notebooks/Custom_derivative_rules_for_Python_code.md
+2 b/‎docs/notebooks/Custom_derivative_rules_for_Python_code.md
+2
diff --git a/‎docs/notebooks/Distributed_arrays_and_automatic_parallelization.ipynb
+3-1 b/‎docs/notebooks/Distributed_arrays_and_automatic_parallelization.ipynb
+3-1
diff --git a/‎docs/notebooks/Distributed_arrays_and_automatic_parallelization.md
+2 b/‎docs/notebooks/Distributed_arrays_and_automatic_parallelization.md
+2
diff --git a/‎docs/notebooks/How_JAX_primitives_work.ipynb
+2 b/‎docs/notebooks/How_JAX_primitives_work.ipynb
+2
diff --git a/‎docs/notebooks/How_JAX_primitives_work.md
+2 b/‎docs/notebooks/How_JAX_primitives_work.md
+2
diff --git a/‎docs/notebooks/Neural_Network_and_Data_Loading.ipynb
+2 b/‎docs/notebooks/Neural_Network_and_Data_Loading.ipynb
+2
@@ -36,6 +36,8 @@
    "source": [
     "# Autodidax: JAX core from scratch\n",
     "\n",
+    "<!--* freshness: { reviewed: '2024-04-08' } *-->\n",
+    "\n",
     "Ever want to learn how JAX works, but the implementation seemed impenetrable?\n",
     "Well, you're in luck! By reading this tutorial, you'll learn every big idea in\n",
     "JAX's core system. You'll even get clued into our weird jargon!\n",
 
@@ -31,6 +31,8 @@
 
 # # Autodidax: JAX core from scratch
 #
+# <!--* freshness: { reviewed: '2024-04-08' } *-->
+#
 # Ever want to learn how JAX works, but the implementation seemed impenetrable?
 # Well, you're in luck! By reading this tutorial, you'll learn every big idea in
 # JAX's core system. You'll even get clued into our weird jargon!
 
@@ -8,6 +8,8 @@
    "source": [
     "# 🔪 JAX - The Sharp Bits 🔪\n",
     "\n",
+    "<!--* freshness: { reviewed: '2024-06-03' } *-->\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/Common_Gotchas_in_JAX.ipynb) [![Open in Kaggle](https://kaggle.com/static/images/open-in-kaggle.svg)](https://kaggle.com/kernels/welcome?src=https://github.com/google/jax/blob/main/docs/notebooks/Common_Gotchas_in_JAX.ipynb)"
    ]
   },
 
@@ -8,6 +8,8 @@
    "source": [
     "# Custom derivative rules for JAX-transformable Python functions\n",
     "\n",
+    "<!--* freshness: { reviewed: '2024-04-08' } *-->\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/Custom_derivative_rules_for_Python_code.ipynb) [![Open in Kaggle](https://kaggle.com/static/images/open-in-kaggle.svg)](https://kaggle.com/kernels/welcome?src=https://github.com/google/jax/blob/main/docs/notebooks/Custom_derivative_rules_for_Python_code.ipynb)\n",
     "\n",
     "*mattjj@ Mar 19 2020, last updated Oct 14 2020*\n",
 
@@ -6,7 +6,9 @@
     "id": "PxHrg4Cjuapm"
    },
    "source": [
-    "# Distributed arrays and automatic parallelization"
+    "# Distributed arrays and automatic parallelization\n",
+    "\n",
+    "<!--* freshness: { reviewed: '2024-04-16' } *-->"
    ]
   },
   {
 
@@ -8,6 +8,8 @@
    "source": [
     "# How JAX primitives work\n",
     "\n",
+    "<!--* freshness: { reviewed: '2024-04-08' } *-->\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/How_JAX_primitives_work.ipynb) [![Open in Kaggle](https://kaggle.com/static/images/open-in-kaggle.svg)](https://kaggle.com/kernels/welcome?src=https://github.com/google/jax/blob/main/docs/notebooks/How_JAX_primitives_work.ipynb)\n",
     "\n",
     "*[email protected]*, October 2019.\n",
 
@@ -8,6 +8,8 @@
    "source": [
     "# Training a Simple Neural Network, with PyTorch Data Loading\n",
     "\n",
+    "<!--* freshness: { reviewed: '2024-05-03' } *-->\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/Neural_Network_and_Data_Loading.ipynb) [![Open in Kaggle](https://kaggle.com/static/images/open-in-kaggle.svg)](https://kaggle.com/kernels/welcome?src=https://github.com/google/jax/blob/main/docs/notebooks/Neural_Network_and_Data_Loading.ipynb)\n",
     "\n",
     "**Copyright 2018 The JAX Authors.**\n",
Original file line number	Diff line number	Diff line change
`@@ -31,6 +31,8 @@`
`31`	`31`
`32`	`32`	`# # Autodidax: JAX core from scratch`
`33`	`33`	`#`
	`34`	`+# <!--* freshness: { reviewed: '2024-04-08' } *-->`
	`35`	`+#`
`34`	`36`	`# Ever want to learn how JAX works, but the implementation seemed impenetrable?`
`35`	`37`	`# Well, you're in luck! By reading this tutorial, you'll learn every big idea in`
`36`	`38`	`# JAX's core system. You'll even get clued into our weird jargon!`
Original file line number	Diff line number	Diff line change
`@@ -8,6 +8,8 @@`
`8`	`8`	`"source": [`
`9`	`9`	`"# 🔪 JAX - The Sharp Bits 🔪\n",`
`10`	`10`	`"\n",`
	`11`	`+ "<!--* freshness: { reviewed: '2024-06-03' } *-->\n",`
	`12`	`+ "\n",`
`11`	`13`	`"[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/jax/blob/main/docs/notebooks/Common_Gotchas_in_JAX.ipynb) [![Open in Kaggle](https://kaggle.com/static/images/open-in-kaggle.svg)](https://kaggle.com/kernels/welcome?src=https://github.com/google/jax/blob/main/docs/notebooks/Common_Gotchas_in_JAX.ipynb)"`
`12`	`14`	`]`
`13`	`15`	`},`
Original file line number	Diff line number	Diff line change
`@@ -6,7 +6,9 @@`
`6`	`6`	`"id": "PxHrg4Cjuapm"`
`7`	`7`	`},`
`8`	`8`	`"source": [`
`9`		`- "# Distributed arrays and automatic parallelization"`
	`9`	`+ "# Distributed arrays and automatic parallelization\n",`
	`10`	`+ "\n",`
	`11`	`+ "<!--* freshness: { reviewed: '2024-04-16' } *-->"`
`10`	`12`	`]`
`11`	`13`	`},`
`12`	`14`	`{`