diff --git a/.docker.env b/.docker.env
deleted file mode 100644
index 961ef54..0000000
--- a/.docker.env
+++ /dev/null
@@ -1,2 +0,0 @@
-DOCKER_BUILDKIT=1
-COMPOSE_DOCKER_CLI_BUILD=1
diff --git a/.github/workflows/docker-image.yml b/.github/workflows/docker-image.yml
index 66b3d5b..ac59b67 100644
--- a/.github/workflows/docker-image.yml
+++ b/.github/workflows/docker-image.yml
@@ -2,18 +2,135 @@ name: Docker Image CI
 
 on:
   push:
-    branches: [ "main" ]
+    branches: ["docker"]
+  release:
+    types: [published]
+
+permissions:
+  contents: read
+  packages: write
 
 jobs:
-  build:
+  build-base:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          submodules: recursive
+
+      - name: Log in to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Build and push dev-base
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          file: docker/Dockerfile.dev
+          target: dev-base
+          platforms: linux/amd64
+          push: true
+          tags: ghcr.io/collaborativebioinformatics/oncolearn:dev-base
+          cache-from: type=gha,scope=dev-base
+          cache-to: type=gha,mode=max,scope=dev-base
+
+      - name: Build and push prod-base
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          file: docker/Dockerfile.prod
+          target: pytorch-prod-base
+          platforms: linux/amd64
+          push: true
+          tags: ghcr.io/collaborativebioinformatics/oncolearn:prod-base
+          cache-from: type=gha,scope=prod-base
+          cache-to: type=gha,mode=max,scope=prod-base
+
+  build-gpu:
+    needs: build-base
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        include:
+          - stage: dev
+            gpu_extra: cpu
+            image_tag: dev-cpu
+            rocm_wsl: "0"
+          - stage: dev
+            gpu_extra: cu130
+            image_tag: dev-cuda
+            rocm_wsl: "0"
+          - stage: dev
+            gpu_extra: rocm
+            image_tag: dev-rocm
+            rocm_wsl: "0"
+          - stage: prod
+            gpu_extra: cpu
+            image_tag: prod-cpu
+            rocm_wsl: "0"
+          - stage: prod
+            gpu_extra: cu130
+            image_tag: prod-cuda
+            rocm_wsl: "0"
+          - stage: prod
+            gpu_extra: rocm
+            image_tag: prod-rocm
+            rocm_wsl: "0"
+          - stage: prod
+            gpu_extra: rocm
+            image_tag: prod-rocm-wsl
+            rocm_wsl: "1"
 
     steps:
-    - uses: actions/checkout@v4
-
-    - name: Build the Docker image (amd-wsl)
-      run: |
-        docker build . \
-          --file Dockerfile \
-          --build-arg GPU_EXTRA=rocm62 \
-          --tag my-image-name:amd-wsl-$(date +%s)
+      - uses: actions/checkout@v4
+        with:
+          submodules: recursive
+
+      - name: Log in to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Set image variables
+        id: vars
+        run: |
+          if [ "${{ matrix.stage }}" = "dev" ]; then
+            echo "dockerfile=docker/Dockerfile.dev" >> "$GITHUB_OUTPUT"
+            echo "target=dev" >> "$GITHUB_OUTPUT"
+            echo "base_scope=dev-base" >> "$GITHUB_OUTPUT"
+            echo "base_image=ghcr.io/collaborativebioinformatics/oncolearn:dev-base" >> "$GITHUB_OUTPUT"
+          else
+            echo "dockerfile=docker/Dockerfile.prod" >> "$GITHUB_OUTPUT"
+            echo "target=pytorch-prod" >> "$GITHUB_OUTPUT"
+            echo "base_scope=prod-base" >> "$GITHUB_OUTPUT"
+            echo "base_image=ghcr.io/collaborativebioinformatics/oncolearn:prod-base" >> "$GITHUB_OUTPUT"
+          fi
+
+      - name: Build and push GPU variant
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          file: ${{ steps.vars.outputs.dockerfile }}
+          target: ${{ steps.vars.outputs.target }}
+          platforms: linux/amd64
+          push: true
+          tags: ghcr.io/collaborativebioinformatics/oncolearn:${{ matrix.image_tag }}
+          build-args: |
+            GPU_EXTRA=${{ matrix.gpu_extra }}
+            ROCM_WSL=${{ matrix.rocm_wsl }}
+          cache-from: |
+            type=gha,scope=${{ matrix.image_tag }}
+            type=gha,scope=${{ steps.vars.outputs.base_scope }}
+            type=registry,ref=${{ steps.vars.outputs.base_image }}
+          cache-to: type=gha,mode=max,scope=${{ matrix.image_tag }}
diff --git a/docker-compose.yml b/docker-compose.yml
index 2841b81..58a3589 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,6 +1,5 @@
 # Shared configuration using YAML anchors
 x-common-config: &common-config
-  env_file: .docker.env
   volumes:
     - .:/workspace
     - uv-cache:/root/.cache/uv
@@ -21,9 +20,10 @@ x-prod-build: &prod-build
   dockerfile: docker/Dockerfile.prod
 
 services:
-  # Main development environment with Python + R (CPU only)
+  # Development environment — CPU only
   dev-cpu:
     <<: *common-config
+    image: ghcr.io/collaborativebioinformatics/oncolearn:dev-cpu
     profiles: ["dev-cpu"]
     build:
       <<: *dev-build
@@ -32,50 +32,54 @@ services:
       target: dev
     container_name: oncolearn-dev-cpu
 
-  # Main development environment with Python + R (NVIDIA GPU)
-  dev-nvidia:
+  # Development environment — NVIDIA GPU
+  dev-cuda:
     <<: *common-config
-    profiles: ["dev-nvidia"]
+    image: ghcr.io/collaborativebioinformatics/oncolearn:dev-cuda
+    profiles: ["dev-cuda"]
     build:
       <<: *dev-build
       args:
         GPU_EXTRA: cu130
       target: dev
-    container_name: oncolearn-dev-nvidia
+    container_name: oncolearn-dev-cuda
     deploy:
       resources:
         reservations:
           devices:
-            - driver: nvidia
+            - driver: cuda
               count: all
               capabilities: [gpu]
 
-  # Main development environment with Python + R (AMD GPU - Native Linux)
-  dev-amd:
+  # Development environment — AMD GPU (native Linux)
+  dev-rocm:
     <<: *common-config
-    profiles: ["dev-amd"]
+    image: ghcr.io/collaborativebioinformatics/oncolearn:dev-rocm
+    profiles: ["dev-rocm"]
     build:
       <<: *dev-build
       args:
         GPU_EXTRA: rocm
       target: dev
-    container_name: oncolearn-dev-amd
+    container_name: oncolearn-dev-rocm
     devices:
       - /dev/kfd
       - /dev/dri
     group_add:
       - video
 
-  # Main development environment with Python + R (AMD GPU - WSL2)
-  dev-amd-wsl:
+  # Development environment — AMD GPU (WSL2)
+  # Uses the same image as dev-rocm; WSL2 differences are runtime-only.
+  dev-rocm-wsl:
     <<: *common-config
-    profiles: ["dev-amd-wsl"]
+    image: ghcr.io/collaborativebioinformatics/oncolearn:dev-rocm
+    profiles: ["dev-rocm-wsl"]
     build:
       <<: *dev-build
       args:
         GPU_EXTRA: rocm
       target: dev
-    container_name: oncolearn-dev-amd-wsl
+    container_name: oncolearn-dev-rocm-wsl
     volumes:
       - .:/workspace
       - uv-cache:/root/.cache/uv
@@ -99,30 +103,30 @@ services:
     ipc: host  # required for ROCm shared memory
     shm_size: '2gb'
 
-  # Production PyTorch environment (CPU only)
-  pytorch-prod-cpu:
-    env_file: .docker.env
+  # Production image — CPU only
+  prod-cpu:
+    image: ghcr.io/collaborativebioinformatics/oncolearn:prod-cpu
     profiles: ["prod-cpu"]
     build:
       <<: *prod-build
       args:
         GPU_EXTRA: cpu
       target: pytorch-prod
-    container_name: oncolearn-pytorch-prod-cpu
+    container_name: oncolearn-prod-cpu
     volumes:
       - ./data:/workspace/data:ro
       - ./data/configs:/workspace/data/configs
 
-  # Production PyTorch environment (NVIDIA GPU - Optimized Size)
-  pytorch-prod-nvidia:
-    env_file: .docker.env
-    profiles: ["prod-nvidia"]
+  # Production image — NVIDIA GPU
+  prod-cuda:
+    image: ghcr.io/collaborativebioinformatics/oncolearn:prod-cuda
+    profiles: ["prod-cuda"]
     build:
       <<: *prod-build
       args:
         GPU_EXTRA: cu130
       target: pytorch-prod
-    container_name: oncolearn-pytorch-prod-nvidia
+    container_name: oncolearn-prod-cuda
     volumes:
       - ./data:/workspace/data:ro
       - ./data/configs:/workspace/data/configs
@@ -130,21 +134,55 @@ services:
       resources:
         reservations:
           devices:
-            - driver: nvidia
+            - driver: cuda
               count: all
               capabilities: [gpu]
 
-  # Production PyTorch environment (AMD GPU - WSL2)
-  pytorch-prod-amd-wsl:
-    env_file: .docker.env
-    profiles: ["prod-amd"]
+  # Production image — AMD GPU (native Linux)
+  prod-rocm:
+    image: ghcr.io/collaborativebioinformatics/oncolearn:prod-rocm
+    profiles: ["prod-rocm"]
+    build:
+      <<: *prod-build
+      args:
+        GPU_EXTRA: rocm
+        ROCM_WSL: "0"
+      target: pytorch-prod
+    container_name: oncolearn-prod-rocm
+    volumes:
+      - ./data:/workspace/data:ro
+      - ./data/configs:/workspace/data/configs
+      - ./models:/workspace/models
+      - ./outputs:/workspace/outputs
+      - ./.hf-cache:/root/.cache/huggingface
+      - ./src:/workspace/src
+      - /opt/rocm/lib:/opt/rocm/lib:ro  # ROCm runtime libs
+    environment:
+      - LD_LIBRARY_PATH=/opt/rocm/lib
+    devices:
+      - /dev/kfd
+      - /dev/dri
+    group_add:
+      - video
+    cap_add:
+      - SYS_PTRACE
+    security_opt:
+      - seccomp:unconfined
+    ipc: host
+    shm_size: '2gb'
+
+  # Production image — AMD GPU (WSL2)
+  # Uses a separate image with ROCm 7.2 LW wheels built for /dev/dxg.
+  prod-rocm-wsl:
+    image: ghcr.io/collaborativebioinformatics/oncolearn:prod-rocm-wsl
+    profiles: ["prod-rocm-wsl"]
     build:
       <<: *prod-build
       args:
         GPU_EXTRA: rocm
         ROCM_WSL: "1"
       target: pytorch-prod
-    container_name: oncolearn-pytorch-prod-amd
+    container_name: oncolearn-prod-rocm-wsl
     mem_limit: 20g        # cap below WSL limit so OOM kills container, not WSL
     memswap_limit: 22g    # allow some swap headroom during model loading
     volumes:
diff --git a/docker/Dockerfile.dev b/docker/Dockerfile.dev
index ed2f483..dc61a08 100644
--- a/docker/Dockerfile.dev
+++ b/docker/Dockerfile.dev
@@ -68,19 +68,18 @@ RUN mkdir -p /opt/pdc-client \
     && rm -rf /tmp/pdc-client.zip /tmp/pdc-client_v1.0.8_Ubuntu_x64 /tmp/__MACOSX
 
 # ============================================================================
-# Stage 4: Python Dependencies Builder
-# Base: python:3.12-slim + only the build tools required to compile native extensions.
-# python:3.12 (full) is ~1 GB; slim + build-essential is ~200 MB.
+# Stage 4a: Python Dependencies Builder (CPU base)
+# Installs all deps with the CPU torch variant. This layer is shared by all
+# GPU variants — GPU builds extend this stage and reinstall torch on top.
 # ============================================================================
-FROM python:3.12-slim AS python-builder
-
-ARG GPU_EXTRA
+FROM python:3.12-slim AS python-builder-base
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV UV_CACHE_DIR=/workspace/.uv-cache
 ENV UV_PROJECT_ENVIRONMENT=/workspace/.venv
 ENV UV_NO_PROGRESS=0
 ENV UV_HTTP_TIMEOUT=300
+ENV UV_LINK_MODE=copy
 ENV VIRTUAL_ENV=/workspace/.venv
 ENV PATH="/workspace/.venv/bin:${PATH}"
 # bmfm-targets uses setuptools_scm; without a git repo in Docker it hangs trying hg
@@ -107,14 +106,26 @@ COPY pyproject.toml uv.lock ./
 # biomed-multi-omic is a local path dependency required by bmfm-targets
 COPY submodules/biomed-multi-omic ./submodules/biomed-multi-omic/
 
-# Install all third-party deps first (--no-install-project skips oncolearn itself).
-# This layer stays cached as long as pyproject.toml/uv.lock/submodule don't change,
-# even when src/ is edited frequently.
+# Install all third-party deps with CPU torch (--no-install-project skips oncolearn itself).
+# This layer stays cached as long as pyproject.toml/uv.lock/submodule don't change.
 RUN --mount=type=cache,target=/workspace/.uv-cache,sharing=locked \
-    uv sync --no-dev --extra ${GPU_EXTRA} --extra multimodal --extra eda --extra image --no-install-project
+    uv sync --no-dev --extra cpu --extra multimodal --extra eda --extra image --no-install-project
 
 # Install oncolearn editable (cheap; only re-runs when src/ changes)
 COPY src/ ./src/
+RUN --mount=type=cache,target=/workspace/.uv-cache,sharing=locked \
+    uv sync --no-dev --extra cpu --extra multimodal --extra eda --extra image
+
+# ============================================================================
+# Stage 4b: Python Dependencies Builder (GPU variant)
+# Extends the CPU base by reinstalling torch with the requested GPU variant.
+# When GPU_EXTRA=cpu this is effectively a no-op.
+# ============================================================================
+FROM python-builder-base AS python-builder-gpu
+
+ARG GPU_EXTRA=cpu
+
+# Reinstall torch with the GPU-specific variant on top of the existing venv.
 RUN --mount=type=cache,target=/workspace/.uv-cache,sharing=locked \
     uv sync --no-dev --extra ${GPU_EXTRA} --extra multimodal --extra eda --extra image
 
@@ -146,14 +157,11 @@ RUN --mount=type=cache,target=/workspace/.renv-cache,sharing=locked \
           renv::restore();"
 
 # ============================================================================
-# Stage 6: Dev Runtime Image
-# Base: ubuntu:22.04 (Jammy slim, ~77 MB) instead of buildpack-deps:jammy (~500 MB).
-# buildpack-deps includes compilers/build tools which are not needed at runtime —
-# all compilation happens in the builder stages above.
+# Stage 6a: Dev Base Image (CPU torch)
+# This is the publishable base image. Identical to the final dev image but
+# carries CPU torch — GPU images copy the GPU venv on top.
 # ============================================================================
-FROM ubuntu:22.04 AS dev
-
-ARG GPU_EXTRA
+FROM ubuntu:22.04 AS dev-base
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
@@ -209,17 +217,17 @@ RUN mkdir -p /usr/share/desktop-directories \
 COPY --from=pdc-builder /opt/pdc-client /opt/pdc-client
 RUN ln -sf /opt/pdc-client/pdc-client /usr/local/bin/pdc_client
 
-# Copy Python 3.12 interpreter from python-builder stage.
+# Copy Python 3.12 interpreter from python-builder-base stage.
 # ubuntu:22.04 does not ship Python 3.12; the venv's bin/python symlinks to
 # /usr/local/bin/python3.12, which must exist in the runtime image.
-COPY --from=python-builder /usr/local/bin/python3.12 /usr/local/bin/python3.12
-COPY --from=python-builder /usr/local/lib/libpython3.12.so.1.0 /usr/local/lib/libpython3.12.so.1.0
-COPY --from=python-builder /usr/local/lib/python3.12 /usr/local/lib/python3.12
+COPY --from=python-builder-base /usr/local/bin/python3.12 /usr/local/bin/python3.12
+COPY --from=python-builder-base /usr/local/lib/libpython3.12.so.1.0 /usr/local/lib/libpython3.12.so.1.0
+COPY --from=python-builder-base /usr/local/lib/python3.12 /usr/local/lib/python3.12
 RUN ldconfig
 
-# Copy Python virtual environment and uv from python-builder stage
-COPY --from=python-builder /workspace/.venv /workspace/.venv
-COPY --from=python-builder /usr/local/bin/uv /usr/local/bin/uv
+# Copy Python virtual environment (CPU torch) and uv from python-builder-base stage
+COPY --from=python-builder-base /workspace/.venv /workspace/.venv
+COPY --from=python-builder-base /usr/local/bin/uv /usr/local/bin/uv
 
 # Copy R installation and libraries from r-builder stage.
 # rocker/r-ver installs R entirely under /usr/local/ (not /usr/lib/R or /usr/bin/).
@@ -233,3 +241,15 @@ WORKDIR /workspace
 COPY src/ ./src/
 COPY scripts/ ./scripts/
 COPY .Rprofile renv.lock ./
+
+# ============================================================================
+# Stage 6b: Dev Runtime Image (GPU variant)
+# Extends dev-base by overwriting the CPU venv with the GPU-specific venv.
+# ============================================================================
+FROM dev-base AS dev
+
+ARG GPU_EXTRA=cpu
+
+# Overwrite CPU venv with GPU-specific venv (no-op for cpu builds since
+# python-builder-gpu is identical to python-builder-base when GPU_EXTRA=cpu)
+COPY --from=python-builder-gpu /workspace/.venv /workspace/.venv
diff --git a/docker/Dockerfile.prod b/docker/Dockerfile.prod
index 9d22e74..f89bb9e 100644
--- a/docker/Dockerfile.prod
+++ b/docker/Dockerfile.prod
@@ -4,14 +4,11 @@
 ARG GPU_EXTRA=cpu
 
 # ============================================================================
-# Stage 1: Python Dependencies Builder
-# Base: python:3.12-slim + only the build tools required to compile native extensions.
-# python:3.12 (full) is ~1 GB; slim + build-essential is ~200 MB.
+# Stage 1a: Python Dependencies Builder (CPU base)
+# Installs all deps with the CPU torch variant. This layer is shared by all
+# GPU variants — GPU builds extend this stage and reinstall torch on top.
 # ============================================================================
-FROM python:3.12-slim AS pytorch-prod-builder
-
-ARG GPU_EXTRA
-ARG ROCM_WSL=0
+FROM python:3.12-slim AS pytorch-prod-builder-base
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV UV_CACHE_DIR=/workspace/.uv-cache
@@ -44,28 +41,35 @@ WORKDIR /workspace
 COPY pyproject.toml uv.lock ./
 COPY submodules/biomed-multi-omic ./submodules/biomed-multi-omic/
 
-# Install all third-party deps first (--no-install-project skips oncolearn itself).
-# This layer stays cached as long as pyproject.toml/uv.lock/submodule don't change,
-# even when src/ is edited frequently.
+# Install all third-party deps with CPU torch (--no-install-project skips oncolearn itself).
+# This layer stays cached as long as pyproject.toml/uv.lock/submodule don't change.
 RUN --mount=type=cache,target=/workspace/.uv-cache,sharing=locked \
-    if [ "$ROCM_WSL" = "1" ]; then \
-      uv sync --no-dev --extra multimodal --no-install-project \
-        --no-install-package torch \
-        --no-install-package torchvision \
-        --no-install-package triton; \
-    else \
-      uv sync --no-dev --extra ${GPU_EXTRA} --extra multimodal --no-install-project; \
-    fi
+    uv sync --no-dev --extra cpu --extra multimodal --no-install-project
 
 # Install oncolearn editable (cheap; only re-runs when src/ changes)
 COPY src/ ./src/
+RUN --mount=type=cache,target=/workspace/.uv-cache,sharing=locked \
+    uv sync --no-dev --extra cpu --extra multimodal
+
+# ============================================================================
+# Stage 1b: Python Dependencies Builder (GPU variant)
+# Extends the CPU base by reinstalling torch with the requested GPU variant.
+# For ROCM_WSL=1 builds, torch is removed here and the ROCm 7.2 LW wheels
+# are installed in the runtime stage instead.
+# ============================================================================
+FROM pytorch-prod-builder-base AS pytorch-prod-builder-gpu
+
+ARG GPU_EXTRA=cpu
+ARG ROCM_WSL=0
+
 RUN --mount=type=cache,target=/workspace/.uv-cache,sharing=locked \
     if [ "$ROCM_WSL" = "1" ]; then \
+      uv pip uninstall torch torchvision triton 2>/dev/null || true; \
       uv sync --no-dev --extra multimodal \
         --no-install-package torch \
         --no-install-package torchvision \
         --no-install-package triton; \
-    else \
+    elif [ "${GPU_EXTRA}" != "cpu" ]; then \
       uv sync --no-dev --extra ${GPU_EXTRA} --extra multimodal; \
     fi
 
@@ -110,13 +114,11 @@ done
 SHELL
 
 # ============================================================================
-# Stage 3: Production Runtime
-# Base: python:3.12-slim (~125 MB) — only runtime libs, no compilers or build tools.
+# Stage 3a: Production Base Image (CPU torch)
+# This is the publishable base image. Copies the CPU venv — GPU images copy
+# the GPU venv on top.
 # ============================================================================
-FROM python:3.12-slim AS pytorch-prod
-
-ARG GPU_EXTRA
-ARG ROCM_WSL=0
+FROM python:3.12-slim AS pytorch-prod-base
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
@@ -136,9 +138,26 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
         libelf1 \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy the built venv and uv (needed only to install ROCm wheels below)
-COPY --from=pytorch-prod-builder /workspace/.venv /workspace/.venv
-COPY --from=pytorch-prod-builder /usr/local/bin/uv /usr/local/bin/uv
+# Copy the built venv (CPU torch) and uv from the base builder
+COPY --from=pytorch-prod-builder-base /workspace/.venv /workspace/.venv
+COPY --from=pytorch-prod-builder-base /usr/local/bin/uv /usr/local/bin/uv
+
+WORKDIR /workspace
+
+COPY src/ ./src/
+
+# ============================================================================
+# Stage 3b: Production Runtime Image (GPU variant)
+# Extends pytorch-prod-base by overwriting the CPU venv with the GPU-specific
+# venv, then installs ROCm wheels for WSL2 AMD builds.
+# ============================================================================
+FROM pytorch-prod-base AS pytorch-prod
+
+ARG GPU_EXTRA=cpu
+ARG ROCM_WSL=0
+
+# Overwrite CPU venv with GPU-specific venv
+COPY --from=pytorch-prod-builder-gpu /workspace/.venv /workspace/.venv
 
 # Copy pre-fetched ROCm wheels (empty dir for non-WSL builds)
 COPY --from=rocm-wsl-wheels /rocm-wheels /rocm-wheels
@@ -154,7 +173,3 @@ RUN if [ "$GPU_EXTRA" = "rocm" ] && [ "$ROCM_WSL" = "1" ]; then \
         /rocm-wheels/triton-*.whl; \
     fi && \
     rm -rf /rocm-wheels /usr/local/bin/uv
-
-WORKDIR /workspace
-
-COPY src/ ./src/
diff --git a/pyproject.toml b/pyproject.toml
index 3c045a3..d8de382 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -87,6 +87,9 @@ dev = [
 ]
 
 [tool.uv]
+required-environments = [
+    "sys_platform == 'linux' and platform_machine == 'x86_64' and platform_python_implementation == 'CPython'",
+]
 conflicts = [
     [
         { extra = "cpu" },
diff --git a/submodules/wiki b/submodules/wiki
index 934b4dd..9d65f39 160000
--- a/submodules/wiki
+++ b/submodules/wiki
@@ -1 +1 @@
-Subproject commit 934b4dd3c9ae05168382cd283650335d2aad901e
+Subproject commit 9d65f395a24cdb4ce473c357c0d3f336cff30c69
diff --git a/uv.lock b/uv.lock
index ec20e0c..7acf9d7 100644
--- a/uv.lock
+++ b/uv.lock
@@ -8,10 +8,14 @@ resolution-markers = [
     "(platform_machine != 'aarch64' and extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm')",
     "platform_machine == 'aarch64' and platform_python_implementation == 'CPython' and sys_platform == 'linux' and extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm'",
     "sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm'",
-    "sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm'",
+    "(platform_machine != 'x86_64' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm')",
+    "platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm'",
     "sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm'",
     "extra != 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm'",
 ]
+required-markers = [
+    "platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and sys_platform == 'linux'",
+]
 conflicts = [[
     { package = "oncolearn", extra = "cpu" },
     { package = "oncolearn", extra = "cu128" },
@@ -1424,10 +1428,11 @@ dependencies = [
     { name = "torch", version = "2.9.1+cu128", source = { registry = "https://download.pytorch.org/whl/cu128" }, marker = "extra == 'extra-9-oncolearn-cu128' or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torch", version = "2.9.1+cu130", source = { registry = "https://download.pytorch.org/whl/cu130" }, marker = "extra == 'extra-9-oncolearn-cu130' or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torch", version = "2.9.1+rocm6.4", source = { registry = "https://download.pytorch.org/whl/rocm6.4" }, marker = "(extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm')" },
-    { name = "torchvision", version = "0.24.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
+    { name = "torchvision", version = "0.24.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (platform_python_implementation != 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1", source = { registry = "https://download.pytorch.org/whl/cu128" }, marker = "(platform_machine == 'aarch64' and platform_python_implementation == 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cu128') or (platform_machine != 'aarch64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'aarch64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'aarch64' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1", source = { registry = "https://download.pytorch.org/whl/cu130" }, marker = "(platform_machine == 'aarch64' and platform_python_implementation == 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'aarch64' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1", source = { registry = "https://pypi.org/simple" }, marker = "(extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm')" },
+    { name = "torchvision", version = "0.24.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1+cu128", source = { registry = "https://download.pytorch.org/whl/cu128" }, marker = "(platform_machine != 'aarch64' and extra == 'extra-9-oncolearn-cu128') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1+cu130", source = { registry = "https://download.pytorch.org/whl/cu130" }, marker = "(platform_machine != 'aarch64' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1+d801a34", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
@@ -2166,7 +2171,8 @@ cli = [
 cpu = [
     { name = "torch", version = "2.9.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torch", version = "2.9.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform != 'darwin' and extra == 'extra-9-oncolearn-cpu') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
-    { name = "torchvision", version = "0.24.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
+    { name = "torchvision", version = "0.24.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (platform_python_implementation != 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
+    { name = "torchvision", version = "0.24.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1+d801a34", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
 ]
 cu128 = [
@@ -3739,10 +3745,11 @@ dependencies = [
     { name = "torch", version = "2.9.1+cu128", source = { registry = "https://download.pytorch.org/whl/cu128" }, marker = "extra == 'extra-9-oncolearn-cu128' or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torch", version = "2.9.1+cu130", source = { registry = "https://download.pytorch.org/whl/cu130" }, marker = "extra == 'extra-9-oncolearn-cu130' or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torch", version = "2.9.1+rocm6.4", source = { registry = "https://download.pytorch.org/whl/rocm6.4" }, marker = "(extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm')" },
-    { name = "torchvision", version = "0.24.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
+    { name = "torchvision", version = "0.24.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (platform_python_implementation != 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1", source = { registry = "https://download.pytorch.org/whl/cu128" }, marker = "(platform_machine == 'aarch64' and platform_python_implementation == 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cu128') or (platform_machine != 'aarch64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'aarch64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'aarch64' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1", source = { registry = "https://download.pytorch.org/whl/cu130" }, marker = "(platform_machine == 'aarch64' and platform_python_implementation == 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'aarch64' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1", source = { registry = "https://pypi.org/simple" }, marker = "(extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra != 'extra-9-oncolearn-cu128' and extra != 'extra-9-oncolearn-cu130' and extra != 'extra-9-oncolearn-rocm')" },
+    { name = "torchvision", version = "0.24.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1+cu128", source = { registry = "https://download.pytorch.org/whl/cu128" }, marker = "(platform_machine != 'aarch64' and extra == 'extra-9-oncolearn-cu128') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1+cu130", source = { registry = "https://download.pytorch.org/whl/cu130" }, marker = "(platform_machine != 'aarch64' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torchvision", version = "0.24.1+d801a34", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform != 'darwin' and sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
@@ -3835,7 +3842,8 @@ version = "2.9.1+cpu"
 source = { registry = "https://download.pytorch.org/whl/cpu" }
 resolution-markers = [
     "sys_platform != 'darwin' and sys_platform != 'linux'",
-    "sys_platform == 'linux'",
+    "(platform_machine != 'x86_64' and sys_platform == 'linux') or (platform_python_implementation != 'CPython' and sys_platform == 'linux')",
+    "platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and sys_platform == 'linux'",
 ]
 dependencies = [
     { name = "filelock", marker = "(sys_platform != 'darwin' and extra == 'extra-9-oncolearn-cpu') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
@@ -4033,14 +4041,14 @@ name = "torchvision"
 version = "0.24.1"
 source = { registry = "https://download.pytorch.org/whl/cpu" }
 resolution-markers = [
-    "sys_platform == 'linux'",
+    "(platform_machine != 'x86_64' and sys_platform == 'linux') or (platform_python_implementation != 'CPython' and sys_platform == 'linux')",
     "sys_platform == 'darwin'",
 ]
 dependencies = [
-    { name = "numpy", marker = "(sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
-    { name = "pillow", marker = "(sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
+    { name = "numpy", marker = "(platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (platform_python_implementation != 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
+    { name = "pillow", marker = "(platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (platform_python_implementation != 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
     { name = "torch", version = "2.9.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform == 'darwin' and extra == 'extra-9-oncolearn-cpu') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
-    { name = "torch", version = "2.9.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
+    { name = "torch", version = "2.9.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (platform_python_implementation != 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
 ]
 wheels = [
     { url = "https://download.pytorch.org/whl/cpu/torchvision-0.24.1-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:4f0956825bab5932dd53394de8e1105ed9182502b367ff0475dbf0e699550612" },
@@ -4111,6 +4119,27 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/4f/2b/ba02e4261369c3798310483028495cf507e6cb3f394f42e4796981ecf3a7/torchvision-0.24.1-cp313-cp313t-win_amd64.whl", hash = "sha256:d83e16d70ea85d2f196d678bfb702c36be7a655b003abed84e465988b6128938", size = 4251604, upload-time = "2025-11-12T15:25:34.069Z" },
 ]
 
+[[package]]
+name = "torchvision"
+version = "0.24.1+cpu"
+source = { registry = "https://download.pytorch.org/whl/cpu" }
+resolution-markers = [
+    "platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and sys_platform == 'linux'",
+]
+dependencies = [
+    { name = "numpy", marker = "(platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
+    { name = "pillow", marker = "(platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
+    { name = "torch", version = "2.9.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(platform_machine == 'x86_64' and platform_python_implementation == 'CPython' and sys_platform == 'linux' and extra == 'extra-9-oncolearn-cpu') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_machine != 'x86_64' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (platform_python_implementation != 'CPython' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (sys_platform != 'linux' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-cu130') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu128' and extra == 'extra-9-oncolearn-rocm') or (extra != 'extra-9-oncolearn-cpu' and extra == 'extra-9-oncolearn-cu130' and extra == 'extra-9-oncolearn-rocm')" },
+]
+wheels = [
+    { url = "https://download.pytorch.org/whl/cpu/torchvision-0.24.1%2Bcpu-cp312-cp312-manylinux_2_28_x86_64.whl", hash = "sha256:48ff40694812292f1f75cddb517f9eef224a2e8388798a0aa191e27bf159fc5d" },
+    { url = "https://download.pytorch.org/whl/cpu/torchvision-0.24.1%2Bcpu-cp312-cp312-win_amd64.whl", hash = "sha256:af904a34b563c302fa4583a4746f3108bafdc5823a67c912d6023a8fe9b01c51" },
+    { url = "https://download.pytorch.org/whl/cpu/torchvision-0.24.1%2Bcpu-cp313-cp313-manylinux_2_28_x86_64.whl", hash = "sha256:19eb1744686a992522bf9ab8bf4b3245c2202e93a39b1fe1fcb2b8d2ca982048" },
+    { url = "https://download.pytorch.org/whl/cpu/torchvision-0.24.1%2Bcpu-cp313-cp313-win_amd64.whl", hash = "sha256:4e2a64198ea3a3efdd4d1283e54418a3816a6943f619e5a33fc8236b6e259539" },
+    { url = "https://download.pytorch.org/whl/cpu/torchvision-0.24.1%2Bcpu-cp313-cp313t-manylinux_2_28_x86_64.whl", hash = "sha256:0a3fecbadc155e7bf378178029215bfcd86f2cf453fbb1d9a474f375b3d475ae" },
+    { url = "https://download.pytorch.org/whl/cpu/torchvision-0.24.1%2Bcpu-cp313-cp313t-win_amd64.whl", hash = "sha256:434347899fed98fb6906d86c00789963d3c22f69e18888aac15ffa012751b256" },
+]
+
 [[package]]
 name = "torchvision"
 version = "0.24.1+cu128"