diff --git a/pyproject.toml b/pyproject.toml
index 25509d0ba..0739361e2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -32,8 +32,8 @@ classifiers = [
 ]
 dependencies = [
     "fastprogress>=1.0.0",
-    "jax>=0.4.12",
-    "jaxlib>=0.4.14",
+    "jax>=0.4.16",
+    "jaxlib>=0.4.16",
     "jaxopt>=0.8",
     "optax>=0.1.7",
     "typing-extensions>=4.4.0",
diff --git a/requirements-doc.txt b/requirements-doc.txt
index e5561ed08..34e1ad6ac 100644
--- a/requirements-doc.txt
+++ b/requirements-doc.txt
@@ -4,8 +4,8 @@ aesara>=2.8.8
 arviz
 flax
 ipython
-jax
-jaxlib
+jax>=0.4.16
+jaxlib>=0.4.16
 jaxopt
 jupytext
 myst_nb
diff --git a/setup.cfg b/setup.cfg
index 0060b8133..51b73dc30 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -18,3 +18,4 @@ python_files=test*.py
 testpaths=tests
 filterwarnings =
     error
+    
\ No newline at end of file