diff --git a/pyproject.toml b/pyproject.toml
index f6bb904..f6decb5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -15,21 +15,21 @@ classifiers = [
 ]
 dependencies = [
     "hydrax@git+https://github.com/vincekurtz/hydrax@v0.0.1",
-    "jax[cuda12]==0.4.35",
-    "flax==0.10.0",
-    "pytest==8.3.3",
-    "ruff==0.7.1",
-    "pre-commit==4.0.1",
-    "matplotlib==3.9.2",
-    "mujoco==3.2.4",
-    "mujoco-mjx==3.2.4",
-    "evosax==0.1.6",
-    "optax==0.2.3",
-    "tensorboard==2.18.0",
-    "tensorboardX==2.6.2",
-    "cloudpickle==3.1.0",
-    "moviepy==1.0.3",
-    "imageio==2.27",
+    "jax[cuda12]",
+    "flax<=0.10.0",
+    "pytest",
+    "ruff",
+    "pre-commit",
+    "matplotlib",
+    "mujoco",
+    "mujoco-mjx",
+    "evosax<=0.1.6",
+    "optax",
+    "tensorboard",
+    "tensorboardX",
+    "cloudpickle",
+    "moviepy",
+    "imageio",
 ]
 
 [tool.ruff]