diff --git a/pyproject.toml b/pyproject.toml
index 47bfcb60..afa75efa 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,8 +36,8 @@ dependencies = [
     "drmaa",
     "gevent",
     "harmonypy",
-    "jax<=0.4.23",
-    "jaxlib<=0.4.23",
+    "jax",
+    "jaxlib",
     "leidenalg",
     "louvain",
     "matplotlib<=3.7.3",