Add Unsloth Docker images (base + Studio) for any NVIDIA GPU host, Ampere through Blackwell

.github/workflows/studio-backend-ci.yml

@@ -30,6 +30,11 @@ on:
       - 'unsloth/**'
       - 'unsloth_cli/**'
       - 'tests/**'
+      # The "Docker JupyterLab/notebook feature validation" step below runs
+      # tests/validate_studio_features.py, which checks docker/jupyter (the
+      # labextension, overrides.json, login branding) and the docker notebook
+      # helpers. Without docker/** here a docker-only change skips that guard.
+      - 'docker/**'
       - 'pyproject.toml'
       - '.github/workflows/studio-backend-ci.yml'
   push:
@@ -238,3 +243,9 @@ jobs:
               echo "::endgroup::"
           done
 
+      - name: Docker JupyterLab/notebook feature validation
+        # Named validate_studio_features.py (not test_*.py) so pytest's default
+        # discovery skips it; run it explicitly here so a regression in the
+        # notebook view, Colab compat, strip, JupyterLab defaults or login
+        # branding fails CI instead of only when someone runs it by hand.
+        run: python tests/validate_studio_features.py

.gitignore

@@ -235,5 +235,7 @@ package-lock.json
 !studio/backend/core/data_recipe/oxc-validator/package-lock.json
 !studio/package-lock.json
 llama.cpp/
+async_task_outputs/
+individual_reviews/
 # Stray "~" dir some tools create from a literal ~ TMPDIR; never part of the repo.
 /~/

docker/.dockerignore

@@ -0,0 +1,37 @@
+**
+!Dockerfile
+!entrypoint.sh
+!smoke_test.py
+!fetch_llama_prebuilt.py
+!supervisord.conf
+!studio_launch.sh
+!unsloth_studio_update.sh
+!unsloth_llama_update.sh
+!unsloth_jupyter_tunnel.sh
+!unsloth_nb_compat.py
+!unsloth_pip_shim.py
+!unsloth_nb_pip_magic.py
+!unsloth_ipython_startup.py
+!unsloth_run.py
+!unsloth_sync_notebooks.sh
+!unsloth_nb_content_sig.py
+!unsloth_nb_view.py
+!unsloth_nb_strip_colab.py
+!unsloth_colab_compat.py
+!jupyter
+!jupyter/unsloth_branding.py
+!jupyter/jupyter_server_config.d
+!jupyter/jupyter_server_config.d/**
+!jupyter/overrides.json
+!jupyter/favicon.ico
+!jupyter/logo.png
+!jupyter/login.html
+!jupyter/install_sloth_stickers.py
+!jupyter/unsloth_labext
+!jupyter/unsloth_labext/package.json
+!jupyter/unsloth_labext/tsconfig.json
+!jupyter/unsloth_labext/.yarnrc.yml
+!jupyter/unsloth_labext/src
+!jupyter/unsloth_labext/src/**
+!jupyter/unsloth_labext/style
+!jupyter/unsloth_labext/style/**

docker/NOTICE

@@ -0,0 +1,41 @@
+Unsloth Docker Studio and JupyterLab image
+==========================================
+
+This directory builds the Unsloth Docker Studio and JupyterLab image. The image
+bundles Unsloth Studio, which is licensed under the GNU Affero General Public
+License v3.0 (see /studio/LICENSE.AGPL-3.0). Unsloth Core is licensed under the
+Apache License 2.0 (see /LICENSE).
+
+
+Additional terms under AGPLv3 Section 7
+---------------------------------------
+
+As permitted by Section 7(b) of the GNU Affero General Public License v3.0, and
+in support of the "Appropriate Legal Notices" requirement for interactive user
+interfaces, the following author attributions and legal notices are designated
+as required Appropriate Legal Notices for this image. If you convey, modify, or
+make the image (or any work based on it) available to users over a network, you
+must keep these notices intact and displayed to those users:
+
+  * The attribution "Built by the Unsloth team".
+  * The copyright line "Copyright 2026-Present the Unsloth team".
+  * The license notice "Licensed under Apache 2.0 and the GNU AGPLv3".
+  * The Unsloth logo and the "Unsloth Dark" theme shown in the JupyterLab top
+    bar and on the loading splash.
+  * The Help > About dialog, including the following links:
+      - Source:  https://github.com/unslothai/unsloth
+      - Website: https://unsloth.ai
+      - License: https://github.com/unslothai/unsloth#license
+      - AGPLv3:  https://www.gnu.org/licenses/agpl-3.0.html
+      - Apache:  https://www.apache.org/licenses/LICENSE-2.0
+
+These notices are displayed on the JupyterLab login page, the Help > About
+dialog, the loading splash and the top bar. They are enforced at build time and
+at runtime by docker/jupyter/unsloth_branding.py (see docker/jupyter/BRANDING.md
+for details). Removing or altering them, whether by editing the build workflow,
+the branding sources or the integrity guard, does not remove this license
+condition.
+
+"Unsloth" and the Unsloth logo are trademarks of the Unsloth team. This NOTICE
+governs copyright attribution under the AGPLv3 and does not grant any trademark
+license.

docker/build.sh

@@ -0,0 +1,67 @@
+#!/usr/bin/env bash
+# Build the unsloth-blackwell image on this B200 host (or any Linux host with Docker).
+# The build host's GPU is NOT used -- nvcc cross-compiles for sm_100 + sm_120.
+#
+# Usage:
+#   ./build.sh                 # builds unsloth-blackwell:latest pinned to unsloth main
+#   TAG=2026.05.1 ./build.sh   # custom tag
+#   UNSLOTH_REF=v2026.5.6 UNSLOTH_ZOO_REF=v2026.5.4 ./build.sh   # pin git refs
+set -euo pipefail
+
+cd "$(dirname "$0")"
+
+IMAGE_NAME="${IMAGE_NAME:-unsloth-blackwell}"
+TAG="${TAG:-latest}"
+CUDA_VERSION="${CUDA_VERSION:-12.8.1}"
+UBUNTU_VERSION="${UBUNTU_VERSION:-24.04}"
+PYTHON_VERSION="${PYTHON_VERSION:-3.12}"
+UNSLOTH_REF="${UNSLOTH_REF:-main}"
+UNSLOTH_ZOO_REF="${UNSLOTH_ZOO_REF:-main}"
+
+# llama.cpp prebuilt: default to the newest unslothai/llama.cpp release, resolved
+# here to a concrete tag so the build-arg changes only when upstream publishes a
+# new release (correct Docker layer caching) and the build stays reproducible.
+# Pin it explicitly for a frozen build: LLAMA_PREBUILT_TAG=b9596-mix-e6f2453 ./build.sh
+resolve_latest_llama_tag() {
+    curl -fsSL -o /dev/null -w '%{url_effective}' \
+        "https://github.com/unslothai/llama.cpp/releases/latest" 2>/dev/null \
+        | sed -n 's#.*/releases/tag/##p'
+}
+if [ -z "${LLAMA_PREBUILT_TAG:-}" ]; then
+    LLAMA_PREBUILT_TAG="$(resolve_latest_llama_tag || true)"
+    if [ -n "$LLAMA_PREBUILT_TAG" ]; then
+        echo "Resolved latest llama.cpp release: ${LLAMA_PREBUILT_TAG}"
+    else
+        LLAMA_PREBUILT_TAG="latest"
+        echo "Could not resolve latest llama.cpp tag here; passing 'latest' (resolved inside the build)"
+    fi
+fi
+
+echo "Building ${IMAGE_NAME}:${TAG}"
+echo "  CUDA           ${CUDA_VERSION}  Ubuntu ${UBUNTU_VERSION}  Python ${PYTHON_VERSION}"
+echo "  unsloth        @${UNSLOTH_REF}"
+echo "  unsloth-zoo    @${UNSLOTH_ZOO_REF}"
+echo "  llama.cpp      ${LLAMA_PREBUILT_TAG}"
+echo "  arch list      8.0;8.6;8.9;9.0;10.0;12.0+PTX"
+echo
+
+DOCKER_BUILDKIT=1 docker build \
+    --progress=plain \
+    --build-arg CUDA_VERSION="${CUDA_VERSION}" \
+    --build-arg UBUNTU_VERSION="${UBUNTU_VERSION}" \
+    --build-arg PYTHON_VERSION="${PYTHON_VERSION}" \
+    --build-arg UNSLOTH_REF="${UNSLOTH_REF}" \
+    --build-arg UNSLOTH_ZOO_REF="${UNSLOTH_ZOO_REF}" \
+    --build-arg LLAMA_PREBUILT_TAG="${LLAMA_PREBUILT_TAG}" \
+    -t "${IMAGE_NAME}:${TAG}" \
+    .
+
+echo
+echo "Built ${IMAGE_NAME}:${TAG}"
+echo
+echo "Smoke test on this host (B200, sm_100):"
+echo "  docker run --rm --gpus all ${IMAGE_NAME}:${TAG} python /workspace/smoke_test.py"
+echo
+echo "Smoke test on an RTX 5090 host (sm_120):"
+echo "  docker pull ${IMAGE_NAME}:${TAG}   # or load .tar"
+echo "  docker run --rm --gpus all ${IMAGE_NAME}:${TAG} python /workspace/smoke_test.py"