Update jax version requirements (#666)

junpenglao · web-flow · commit 40efb6cbc92a · 2024-05-08T17:33:59.000+02:00
Fix #665
diff --git a/requirements-doc.txt b/requirements-doc.txt
@@ -4,8 +4,8 @@ aesara>=2.8.8
 arviz
 flax
 ipython
-jax>=0.4.16
-jaxlib>=0.4.16
+jax>=0.4.25
+jaxlib>=0.4.25
 jaxopt
 jupytext
 myst_nb>=1.0.0
