diff --git a/.github/workflows/BuildDockerImages.yml b/.github/workflows/BuildDockerImages.yml
index e229609fe..e263169ab 100644
--- a/.github/workflows/BuildDockerImages.yml
+++ b/.github/workflows/BuildDockerImages.yml
@@ -28,7 +28,7 @@ on:
 
 jobs:
   build_images:
-    runs-on: ubuntu-latest
+    runs-on: ["self-hosted", "tpu", "${{ matrix.device-type }}"]
     steps:
     - uses: actions/checkout@v3
     - name: build jax stable image