From 40efb6cbc92a8074f8ac3a5d9710071f8d987537 Mon Sep 17 00:00:00 2001
From: Junpeng Lao <junpenglao@gmail.com>
Date: Wed, 8 May 2024 17:33:59 +0200
Subject: [PATCH] Update jax version requirements (#666)

Fix https://github.com/blackjax-devs/blackjax/issues/665
---
 requirements-doc.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements-doc.txt b/requirements-doc.txt
index 338073a88..83af1ffe3 100644
--- 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