diff --git a/pyproject.toml b/pyproject.toml
index f3032b6..31ce948 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,8 +20,8 @@ classifiers = [
 keywords = ["parallel", "mapreduce", "machine", "learning", "jax"]
 dependencies = [
     'absl-py>=1.0,<3.0',
-    'jax>=0.4.24,<0.6.0',
-    'jaxlib>=0.4.24,<0.6.0',
+    'jax>=0.4.24',
+    'jaxlib>=0.4.24',
 ]
 dynamic = ['version']
 
diff --git a/requirements.txt b/requirements.txt
index 0057b04..aa6f4be 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 absl-py>=1.0,<3.0
-jax>=0.4.24,<0.6.0
-jaxlib>=0.4.24,<0.6.0
+jax>=0.4.24
+jaxlib>=0.4.24
 chex>=0.1.81