diff --git a/pyproject.toml b/pyproject.toml
index cf63331c..fb05733b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,7 +16,7 @@ dependencies = [
   "neuraltrain==0.0.2",
   "exca==0.5.20",
   "torch>=2.5.1,<2.7",
-  "numpy==2.2.6",
+  "numpy>=2.2.6,<3.0",
   "torchvision>=0.20,<0.22",
   "x_transformers==1.27.20",
   "einops",
@@ -27,10 +27,13 @@ dependencies = [
   "langdetect",
   "spacy",
   "soundfile",
-  "pip",
   "Levenshtein",
   "julius",
-  "transformers"
+  "transformers>=4.40.0",
+  "pandas>=2.2.0",
+  "pydantic>=2.5.0",
+  "requests>=2.31.0",
+  "tqdm>=4.66.0",
 ]
 
 [project.urls]
@@ -50,12 +53,31 @@ Repository = "https://github.com/facebookresearch/tribev2"
   ]
   training = [
     "nibabel",
-    "torchmetrics",
-    "wandb",
-    "lightning",  
+    "torchmetrics>=1.0.0",
+    "wandb>=0.16.0",
+    "lightning>=2.5.0",
   ]
   test = [
     "pytest",
+    "pytest-cov",
+  ]
+  transcription = [
+    "uvx",
+  ]
+  datasets = [
+    "datalad>=0.19.5",
+    "openneuro-py",
+    "nltk==3.8.1",
+    "nltk_contrib @ git+https://github.com/nltk/nltk_contrib.git@683961c53f0c122b90fe2d039fe795e0a2b3e997",
+  ]
+  demo = [
+    "gtts",
+    "langdetect",
+    "spacy",
+    "soundfile",
+  ]
+  all = [
+    "tribev2[plotting,training,test,transcription,datasets,demo]",
   ]
 
 [tool.black]
diff --git a/tribev2/grids/__init__.py b/tribev2/grids/__init__.py
index e69de29b..d6897cd1 100644
--- a/tribev2/grids/__init__.py
+++ b/tribev2/grids/__init__.py
@@ -0,0 +1,12 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+"""Grid configurations and runners for TRIBE v2 experiments."""
+
+from .configs import base_config, mini_config
+from .defaults import default_config
+
+__all__ = ["base_config", "mini_config", "default_config"]