Switch CI to self-hosted runners and parallelize tests to fix TF CPU timeouts

.github/workflows/actions.yml

@@ -29,28 +29,25 @@ jobs:
             version: keras-nightly
           - backend: openvino
             version: keras-nightly
-    runs-on: ubuntu-latest
+
+    container:
+      image: python:3.11-slim
+
+    runs-on: linux-x86-n2-16
+    timeout-minutes: 120
     env:
       KERAS_BACKEND: ${{ matrix.backend }}
     steps:
+    - name: Install binary dependencies
+      # build-essential for C extensions (tokenizers, torch)
+      # curl, git, gnupg for checkout and other actions
+      run: |
+        apt-get update
+        apt-get -y install build-essential curl git gnupg
+        git config --global --add safe.directory '*'
     - uses: actions/checkout@9c091bb21b7c1c1d1991bb908d89e4e9dddfe3e0 # v7.0.0
       with:
         persist-credentials: false
-    - name: Set up Python 3.11
-      uses: actions/setup-python@ece7cb06caefa5fff74198d8649806c4678c61a1 # v6
-      with:
-        python-version: '3.11'
-    - name: Get pip cache dir
-      id: pip-cache
-      run: |
-        python -m pip install --upgrade pip setuptools
-        echo "::set-output name=dir::$(pip cache dir)"
-    - name: pip cache
-      # zizmor: ignore[cache-poisoning]
-      uses: actions/cache@55cc8345863c7cc4c66a329aec7e433d2d1c52a9 # v6.1.0
-      with:
-        path: ${{ steps.pip-cache.outputs.dir }}
-        key: ${{ runner.os }}-pip-${{ hashFiles('requirements-common.txt') }}-${{ hashFiles('requirements.txt') }}-${{ hashFiles('pyproject.toml') }}
     - name: Install dependencies
       run: |
           pip install -r requirements.txt --progress-bar off
@@ -68,7 +65,7 @@ jobs:
         pip install keras-nightly --no-cache-dir --progress-bar off
     - name: Test with pytest
       run: |
-        pytest keras_hub/
+        pytest keras_hub/ -n auto --dist loadfile --durations=50
     - name: Run integration tests
       run: |
         python pip_build.py --install

requirements-common.txt

@@ -13,6 +13,7 @@ packaging
 ruff
 pytest
 pytest-cov
+pytest-xdist
 build
 namex
 # Optional deps.