diff --git a/pyproject.toml b/pyproject.toml
index 95f09a3..19b75ac 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,6 @@ classifiers = [
 keywords = ["gui", "automation", "evaluation", "benchmark", "agent", "ai"]
 
 dependencies = [
-    "open-clip-torch>=2.20.0",
     "pillow>=10.0.0",
     "pydantic-settings>=2.0.0",
     "python-dotenv>=1.2.1",
@@ -86,6 +85,10 @@ wandb = [
 training = [
     # Perceptual hashing for planner cache (falls back to MD5 if not installed)
     "imagehash>=4.3.0",
+    # CLIP embeddings for visual similarity
+    "open-clip-torch>=2.20.0",
+    # Heavy ML deps for openadapt-ml training
+    "openadapt-ml[training]>=0.11.0",
 ]
 verl = [
     # veRL: Volcano Engine Reinforcement Learning for LLMs