diff --git a/pyproject.toml b/pyproject.toml
index b25dd4d7..f521f4b7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -53,7 +53,7 @@ dependencies = [
     "dgl>=2.0.0",
     "pymatgen",
     "lightning",
-    "torch<=2.2.1",
+    "torch",
     "pydantic"
 ]
 version = "1.1.2"