diff --git a/pyproject.toml b/pyproject.toml
index a007454..d5f3793 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -38,19 +38,19 @@ dependencies = [
 
 [project.optional-dependencies]
 dev = [
-    "ruff==0.1.6",
-    "pre-commit==3.3.3"
+    "ruff>=0.1.6",
+    "pre-commit>=3.3.3"
 ]
 
 baselines = [
-    "matplotlib",
-    "imageio",
-    "wandb",
-    "pyrallis",
-    "distrax",
-    "optax",
-    "tensorboard",
-    "orbax"
+    "matplotlib>=3.7.2",
+    "imageio>=2.31.2",
+    "imageio-ffmpeg>=0.4.9",
+    "wandb>=0.15.10",
+    "pyrallis>=0.3.1",
+    "distrax>=0.1.4",
+    "optax>=0.1.5",
+    "orbax>=0.1.9"
 ]