diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index d0fd1c3..e180e59 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -75,7 +75,7 @@ jobs:
         run: uv python install ${{ matrix.python-version }}
 
       - name: Install the project
-        run: uv sync --extra all --group test --resolution lowest-direct
+        run: uv sync --extra all --group test --resolution lowest
 
       - name: Test package
         run: >-
diff --git a/pyproject.toml b/pyproject.toml
index d12a771..1b822a9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,11 +23,11 @@
     "Typing :: Typed",
   ]
   dependencies = [
-    "jax>0.4",
-    "jaxlib>0.4",
+    "jax>=0.4.3",
+    "jaxlib>=0.4.3",
     "jaxtyping>=0.2.34",
     "plum-dispatch>=2.5.2",
-    "quax>0.0.4",
+    "quax>=0.0.5",
   ]
 
   [project.urls]
@@ -175,3 +175,13 @@
 
   [tool.ruff.lint.pydocstyle]
     convention = "numpy"
+
+
+[tool.uv]
+constraint-dependencies = [
+  "bleach>6.0",
+  "pyparsing>=3.0.0",
+  "pyzmq>=25.0",
+  "scipy>=1.13",
+  "jax<0.4.36", # TODO: remove when https://github.com/patrick-kidger/quax/pull/37 is merged
+]