Update nightly uv pipeline

.github/actions/bootstrap-cudnn-ci/action.yml

@@ -0,0 +1,51 @@
+name: Bootstrap cuDNN CI container
+description: Install OS dependencies and uv in CUDA cuDNN container jobs
+inputs:
+  python-version:
+    description: Python major.minor expected in the container
+    required: false
+    default: "3.12"
+runs:
+  using: composite
+  steps:
+    - name: Install system dependencies
+      shell: bash
+      run: |
+        set -euo pipefail
+        export DEBIAN_FRONTEND=noninteractive
+        apt-get update
+        apt-get install -y --no-install-recommends \
+          ca-certificates \
+          curl \
+          git \
+          gh \
+          build-essential \
+          cmake \
+          pkg-config \
+          python3 \
+          python3-dev \
+          python3-venv \
+          python3-pip
+        ln -sf /usr/bin/python3 /usr/bin/python
+        rm -rf /var/lib/apt/lists/*
+
+    - name: Install uv
+      shell: bash
+      run: |
+        set -euo pipefail
+        curl -LsSf https://astral.sh/uv/install.sh | sh
+        echo "$HOME/.cargo/bin" >> "$GITHUB_PATH"
+
+    - name: Print toolchain versions
+      shell: bash
+      run: |
+        set -euo pipefail
+        python3 --version
+        uv --version
+        gcc --version | head -n 1
+        cmake --version | head -n 1
+        if command -v nvcc >/dev/null 2>&1; then
+          nvcc --version
+        else
+          echo "nvcc not found on PATH"
+        fi

.github/actions/build-ci-container/action.yml

@@ -0,0 +1,59 @@
+name: Build CI container image
+description: Build the repository CI container image from Dockerfile
+inputs:
+  image-tag:
+    description: Fully qualified local image tag to build
+    required: false
+    default: physicsnemo-ci:nightly
+  dockerfile:
+    description: Dockerfile path
+    required: false
+    default: Dockerfile
+  context:
+    description: Docker build context path
+    required: false
+    default: .
+  target:
+    description: Docker build target stage
+    required: false
+    default: ci
+  platform:
+    description: Docker target platform
+    required: false
+    default: linux/amd64
+outputs:
+  image-tag:
+    description: The built image tag
+    value: ${{ inputs.image-tag }}
+runs:
+  using: composite
+  steps:
+    - name: Verify Docker CLI availability
+      shell: bash
+      run: |
+        set -euo pipefail
+        docker version
+
+    - name: Build CI container image
+      shell: bash
+      run: |
+        set -euo pipefail
+        echo "::group::docker build"
+        echo "dockerfile=${{ inputs.dockerfile }}"
+        echo "context=${{ inputs.context }}"
+        echo "target=${{ inputs.target }}"
+        echo "platform=${{ inputs.platform }}"
+        echo "tag=${{ inputs.image-tag }}"
+        docker build \
+          --platform "${{ inputs.platform }}" \
+          --file "${{ inputs.dockerfile }}" \
+          --target "${{ inputs.target }}" \
+          --tag "${{ inputs.image-tag }}" \
+          "${{ inputs.context }}"
+        echo "::endgroup::"
+
+    - name: Show built image details
+      shell: bash
+      run: |
+        set -euo pipefail
+        docker image inspect "${{ inputs.image-tag }}" --format='id={{.Id}} size={{.Size}}'

.github/actions/setup-uv-env/action.yml

@@ -0,0 +1,69 @@
+name: Setup uv environment
+description: Restore uv and venv caches, and run uv sync on cache miss
+inputs:
+  uv-cache-key-prefix:
+    description: Prefix for uv package cache key
+    required: true
+  venv-cache-key-prefix:
+    description: Prefix for virtual environment cache key
+    required: true
+  cache-key-suffix:
+    description: Deterministic suffix appended to cache key prefixes
+    required: true
+outputs:
+  uv_cache_hit:
+    description: Whether uv package cache had an exact key hit
+    value: ${{ steps.restore-uv-cache.outputs.cache-hit }}
+  venv_cache_hit:
+    description: Whether venv cache had an exact key hit
+    value: ${{ steps.restore-venv-cache.outputs.cache-hit }}
+runs:
+  using: composite
+  steps:
+    - name: Restore uv package cache
+      id: restore-uv-cache
+      uses: actions/cache/restore@v4
+      with:
+        path: ~/.cache/uv
+        key: ${{ inputs.uv-cache-key-prefix }}-${{ inputs.cache-key-suffix }}
+        fail-on-cache-miss: false
+        restore-keys: |
+          ${{ inputs.uv-cache-key-prefix }}-
+
+    - name: Restore venv cache
+      id: restore-venv-cache
+      uses: actions/cache/restore@v4
+      with:
+        path: .venv
+        key: ${{ inputs.venv-cache-key-prefix }}-${{ inputs.cache-key-suffix }}
+        fail-on-cache-miss: false
+        restore-keys: |
+          ${{ inputs.venv-cache-key-prefix }}-
+
+    - name: Debug cache and environment context
+      shell: bash
+      run: |
+        set -euo pipefail
+        echo "::group::setup-uv-env debug context"
+        echo "uv cache key: ${{ inputs.uv-cache-key-prefix }}-${{ inputs.cache-key-suffix }}"
+        echo "venv cache key: ${{ inputs.venv-cache-key-prefix }}-${{ inputs.cache-key-suffix }}"
+        echo "uv cache exact hit: ${{ steps.restore-uv-cache.outputs.cache-hit }}"
+        echo "venv cache exact hit: ${{ steps.restore-venv-cache.outputs.cache-hit }}"
+        echo "workspace: $GITHUB_WORKSPACE"
+        df -h
+        echo "::endgroup::"
+
+    - name: Install dependencies with uv (dev + cu12)
+      if: steps.restore-venv-cache.outputs.cache-hit != 'true'
+      shell: bash
+      run: |
+        set -euo pipefail
+        export UV_LINK_MODE=copy
+        echo "::group::uv sync (dev + cu12)"
+        uv sync \
+          --frozen \
+          --group dev \
+          --extra cu12
+        echo "::endgroup::"
+        uv run python -c "import torch; print(f'torch={torch.__version__} cuda={torch.version.cuda}')"
+