diff --git a/smartsim/_core/_install/buildenv.py b/smartsim/_core/_install/buildenv.py
index 844d86f4d..c147ee446 100644
--- a/smartsim/_core/_install/buildenv.py
+++ b/smartsim/_core/_install/buildenv.py
@@ -211,6 +211,7 @@ def ml_extras_required(self) -> t.Dict[str, t.List[str]]:
         """Optional ML/DL dependencies we suggest for the user.
         """
         ml_defaults = {
+            "torch": self.TORCH,
             "tensorflow": self.TENSORFLOW,
             "onnx": self.ONNX,
             "skl2onnx": "1.16.0",