Update jax version requirements (#666)

Fix #665
blackjax-devs · May 8, 2024 · 40efb6c · 40efb6c
1 parent 3f92393
commit 40efb6c
Showing 1 changed file with 2 additions and 2 deletions.
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