Merge branch 'main' into nvidia-devcontainer-args

.devcontainer/cuda/devcontainer.json

@@ -3,8 +3,11 @@
     "image": "ghcr.io/luminal-ai/luminal-docker:cuda",
     "initializeCommand": "touch .env",
     "runArgs": [
-        "--gpus=all",
-        "--env-file", ".env"
+        "--env-file",
+        ".env",
+        "--runtime=nvidia",
+        "--env=NVIDIA_VISIBLE_DEVICES=nvidia.com/gpu=all",
+        "--env=NVIDIA_DRIVER_CAPABILITIES=compute,utility"
     ],
     "containerEnv": {
         "CARGO_HOME": "/home/ubuntu/.cache/luminal/cargo"
@@ -49,4 +52,4 @@
             ]
         }
     }
-}
+}

.gitignore

@@ -17,6 +17,7 @@ Cargo.lock
 
 .claude-project
 .claude-memory
+.codex
 
 *.pftrace
 *.safetensors

crates/luminal_python/pyproject.toml

@@ -12,6 +12,16 @@ dependencies = [
     "safetensors",
 ]
 
+[[tool.uv.index]]
+name = "pytorch-cu128"
+url = "https://download.pytorch.org/whl/cu128"
+explicit = true
+
+[tool.uv.sources]
+torch = [
+  { index = "pytorch-cu128", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
+
 
 [build-system]
 requires = ["maturin>=1.0,<2.0"]