diff --git a/pyproject.toml b/pyproject.toml
index 645b8e8..ce1482c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,7 +12,7 @@ keywords = ["scRNAseq", "transformer", "GRN", "gene regulatory network", "scPRIN
 dependencies = [
     "lamindb==0.76.12",
     "cellxgene-census>=0.1.0",
-    "torch==2.2.0",
+    "torch>=2.2.0",
     "lightning>=2.0.0",
     "anndata>=0.9.1",
     "matplotlib>=3.5.0",