add plot styles, clean up package-wide docstrings

pyproject.toml

@@ -24,11 +24,11 @@ classifiers = [
     "Intended Audience :: End Users/Desktop",
 ]
 dependencies = [
+    "torch",
     "colorama>=0.4.6",
     "matplotlib>=3.10.8",
     "numpy>=2.4.1",
     "tensorboard>=2.20.0",
-    "torch>=2.5.1",
     "tqdm>=4.67.1",
 ]
 
@@ -82,3 +82,11 @@ force-sort-within-sections = false
 quote-style = "double"
 indent-style = "space"
 docstring-code-format = true
+
+[tool.uv.sources]
+torch = { index = "pytorch" }
+
+[[tool.uv.index]]
+name = "pytorch"
+url = "https://download.pytorch.org/whl/cu128"
+explicit = true