diff --git a/docs/report/000.ipynb b/docs/report/000.ipynb
index 5e783c4a..f326c088 100644
--- a/docs/report/000.ipynb
+++ b/docs/report/000.ipynb
@@ -54,7 +54,7 @@
    },
    "outputs": [],
    "source": [
-    "%pip install -q black==21.5b2 jax==0.2.13 jaxlib==0.1.67 numpy==1.23 sympy==1.8"
+    "%pip install -q black==21.5b2 jax==0.4.28 jaxlib==0.4.28 numpy==1.23 sympy==1.8"
    ]
   },
   {
diff --git a/docs/report/001.ipynb b/docs/report/001.ipynb
index bb23e434..fb467f3d 100644
--- a/docs/report/001.ipynb
+++ b/docs/report/001.ipynb
@@ -46,7 +46,7 @@
    },
    "outputs": [],
    "source": [
-    "%pip install -q black==21.5b2 jax==0.2.13 jaxlib==0.1.67 matplotlib==3.4.2 numpy==1.19.5 sympy==1.8"
+    "%pip install -q black==21.5b2 jax==0.4.28 jaxlib==0.4.28 matplotlib==3.4.2 numpy==1.26.4 sympy==1.8"
    ]
   },
   {