diff --git a/docs/requirements.txt b/docs/requirements.txt
index d6e7a69..ce79f43 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -15,4 +15,4 @@ sphinx-autodoc-typehints>=3.2.0
 myst_nb>=1.3.0
 sphinx-copybutton>=0.5.2
 
-jax>=9.0.0
\ No newline at end of file
+jax>=0.9.0
\ No newline at end of file