diff --git a/pyproject.toml b/pyproject.toml
index 68056e0..d092b25 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -28,7 +28,6 @@ dependencies = [
     "setuptools>=75.0.0,<76",
     "numpy>=1.26.3,<2",
     "pandas~=2.2.0",
-    "torch==2.5.1; sys_platform == 'darwin'",
     "pyarrow>=16.0.0",
     "joblib>=1.3.0,<2",
     "psutil>=5.9.5,<6",  # upgrade when colab psutil is updated
@@ -41,17 +40,17 @@ dependencies = [
     "formatron>=0.4.11,<0.5",
     "json-repair>=0.30.0,<0.31",
     "rich>=13.9.4,<14",
+    "torch>=2.5.1; sys_platform == 'darwin'",
 ]
 
 [project.optional-dependencies]
 gpu = [
-    "torch==2.5.1",
+    "torch>=2.5.1",
     "bitsandbytes>=0.43.3,<0.44",
     "vllm>=0.6.4,<0.7",
 ]
 cpu = [
-    "torch==2.5.1+cpu; sys_platform != 'darwin'",
-    "torch==2.5.1; sys_platform == 'darwin'"
+    "torch>=2.5.1"
 ]
 
 [dependency-groups]
@@ -93,7 +92,7 @@ explicit = true
 
 [tool.uv.sources]
 torch = [
-  { index = "pytorch-cpu", extra = "cpu", marker = "sys_platform != 'darwin'" },
+  { index = "pytorch-cpu", extra = "cpu"},
 ]
 
 [tool.hatch.build.targets.sdist]