diff --git a/requirements.txt b/requirements.txt
index 796776b51..c5fd26788 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ mkdocs-literate-nav>=0.4.1
 mkdocstrings>=0.19.0
 mkdocstrings-python>=0.7.0
 pygments>=2.12
-jax>=0.4.1
+jax>=0.4.0
 jaxlib>=0.4.1
 pymbar>=4.0.0
 tqdm