diff --git a/pyproject.toml b/pyproject.toml
index 780121a81..990f70846 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -18,6 +18,7 @@ python-dateutil = "^2.8.2"
 annotated-types = "^0.6.0"
 typing-extensions = "^4.8.0"
 lxml = "^4.9.3"
+torch = "^2.1.1"
 
 [tool.poetry.group.dev.dependencies]
 pytest-cov = "^4.1.0"