diff --git a/pyproject.toml b/pyproject.toml
index f2300a02..40dae190 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,7 +10,7 @@ version = "0.0.1"
 # The following assumes that we build on top of the NGC PyTorch Docker image.
 dependencies = [
     # MANDATORY for cneuromax/
-    "torch==2.4.1+cu121", # Tensor manipulation
+    "torch==2.5.1", # Tensor manipulation
     "hydra-core==1.3.2", # Configuration management
     "hydra-zen==0.13.0", # Cleaner Hydra configs
     "beartype==0.19.0", # Dynamic type checking