diff --git a/requirements-doc.txt b/requirements-doc.txt
index 8b0a5a6a..a4d94bdc 100644
--- a/requirements-doc.txt
+++ b/requirements-doc.txt
@@ -1,6 +1,6 @@
 tqdm
-jax
-jaxlib
+jax<0.4.24
+jaxlib<0.4.24
 matplotlib
 numpy
 scipy