huge changes

pyproject.toml

@@ -5,14 +5,27 @@ description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.13"
 dependencies = [
-    "colour-science>=0.4.6",
-    "imageio>=2.37.0",
-    "jupyter>=1.1.1",
-    "jupyterlab>=4.4.3",
-    "numpy>=2.2.6",
-    "pillow>=11.2.1",
-    "pyfftw>=0.15.0",
-    "rawpy>=0.25.0",
-    "scipy>=1.15.3",
-    "warp-lang>=1.7.2",
+  "colour-science>=0.4.6",
+  "imageio>=2.37.0",
+  "jupyter>=1.1.1",
+  "jupyterlab>=4.4.3",
+  "matplotlib>=3.10.3",
+  "numpy>=2.2.6",
+  "opencv-python>=4.11.0.86",
+  "pillow>=11.2.1",
+  "pyfftw>=0.15.0",
+  "rawpy>=0.25.0",
+  "scikit-image>=0.25.2",
+  "scipy>=1.15.3",
+  "torch>=2.7.1",
+  "warp-lang>=1.7.2",
 ]
+
+
+[tool.uv.sources]
+torch = [{ index = "pytorch-cu128" }]
+
+[[tool.uv.index]]
+name = "pytorch-cu128"
+url = "https://download.pytorch.org/whl/cu128"
+explicit = true