diff --git a/requirements/requirements.txt b/requirements/requirements.txt
index 6d56183..9f191cb 100644
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@@ -16,8 +16,8 @@ absl-py
 arviz==0.11.2
 dataclasses;python_version<'3.7'
 immutabledict>=2.0.0
-jax>=0.3.14
-jaxlib>=0.3.14
+jax>=0.3.18
+jaxlib>=0.3.18
 matplotlib==3.3.4
 numpy>=1.12
 numpyro>=0.9.2