diff --git a/envs/env-py310.yml b/envs/env-py310.yml
index c3564a0..b6a9fc0 100644
--- a/envs/env-py310.yml
+++ b/envs/env-py310.yml
@@ -170,3 +170,6 @@ dependencies:
   - line_profiler
   - pyinstrument
   - pyperformance
+  # ML:
+  - gpytorch
+  - pytorch
diff --git a/envs/env-py39.yml b/envs/env-py39.yml
index f9456a2..042c828 100644
--- a/envs/env-py39.yml
+++ b/envs/env-py39.yml
@@ -170,3 +170,6 @@ dependencies:
   - line_profiler
   - pyinstrument
   - pyperformance
+  # ML:
+  - gpytorch
+  - pytorch