diff --git a/.github/workflows/ci-build-checks.yaml b/.github/workflows/ci-build-checks.yaml
index 247b3c82b..9dcdab972 100644
--- a/.github/workflows/ci-build-checks.yaml
+++ b/.github/workflows/ci-build-checks.yaml
@@ -45,8 +45,6 @@ jobs:
           cache: pip
       - name: Install Bazel on CI
         run: ./scripts/ci_install.sh
-      - name: Configure CI TF
-        run: echo "Y\n" | ./configure.sh
       - name: Build Wheel Test
         run: ./scripts/build_pip_package_test.sh
       - name: Test Wheel
@@ -54,13 +52,19 @@ jobs:
 
   bazel-tests:
     name: Library tests
-    runs-on: ubuntu-22.04
+    runs-on: linux-x86-n2-32
+    container:
+      image: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:latest
     timeout-minutes: 60
 
     steps:
+      - name: Adjust the ml-build runner to our needs
+        run: |
+          # lsb-release (needed by setup-python) is missing in ml-build runners
+          sudo apt-get update && sudo apt-get install -y lsb-release
+          # This file, though it's empty, hangs the Bazel install step.
+          rm -f /etc/bazel.bazelrc
       - uses: actions/checkout@1af3b93b6815bc44a9784bd300feb67ff0d1eeb3 # v6.0.0
-      - name: Remove some unnecessary software to free up disk space
-        run: sudo rm -rf /usr/share/dotnet /usr/local/lib/android &
       - uses: actions/setup-python@e797f83bcb11b83ae66e0230d6156d7c80228e7c # v5
         with:
           python-version: '3.10'
@@ -97,11 +101,18 @@ jobs:
 
   tutorials-test:
     name: Tutorial tests
-    runs-on: ubuntu-22.04
-    needs: wheel-build
+    runs-on: linux-x86-n2-32
+    container:
+      image: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:latest
     timeout-minutes: 60
 
     steps:
+      - name: Adjust the ml-build runner to our needs
+        run: |
+          # lsb-release (needed by setup-python) is missing in ml-build runners
+          sudo apt-get update && sudo apt-get install -y lsb-release
+          # This file, though it's empty, hangs the Bazel install step.
+          rm -f /etc/bazel.bazelrc
       - uses: actions/checkout@1af3b93b6815bc44a9784bd300feb67ff0d1eeb3 # v6.0.0
       - uses: actions/setup-python@e797f83bcb11b83ae66e0230d6156d7c80228e7c # v5
         with:
@@ -112,8 +123,6 @@ jobs:
         run: pip install --upgrade pip seaborn==0.10.0
       - name: Install Bazel on CI
         run: ./scripts/ci_install.sh
-      - name: Configure CI TF
-        run: echo "Y\n" | ./configure.sh
       - name: Build Wheel
         run: ./scripts/build_pip_package_test.sh
       - name: Test Notebooks