Add TPU v7 runners to nightly and continuous job

Use the new TPU v7 runners (4 chips, 8 cores) in nightly and continuous jobs.

PiperOrigin-RevId: 842912906

Author: quoctruong

.github/actionlint.yaml

@@ -16,4 +16,5 @@ self-hosted-runner:
     - "linux-x86-ct6e-180-8tpu" # Linux X86 TPU runner using ct6e-hightpu-8t machine with 2x4 topology.
     - "linux-x86-ct6e-180-4tpu" # Linux X86 TPU runner using ct6e-hightpu-4t machine with 2x2 topology.
     - "linux-x86-ct4p-240-4tpu" # Linux X86 TPU runner using ct4p-hightpu-4t machine with 2x2x1 topology.
+    - "linux-x86-tpu7x-224-4tpu" # Linux X86 TPU runner using tpu7x-224 machine with 4 TPU chips (8 cores) and 2x2x1 topology.
     - "linux-x86_64-cirrascale-64-8gpu-amd-mi250" # AMD runner

.github/workflows/cloud-tpu-ci-nightly.yml

@@ -35,7 +35,8 @@ jobs:
         tpu: [
           {type: "v4-8", cores: "4", runner: "linux-x86-ct4p-240-4tpu"},
           {type: "v5e-8", cores: "8", runner: "linux-x86-ct5lp-224-8tpu"},
-          {type: "v6e-8", cores: "8", runner: "linux-x86-ct6e-180-8tpu"}
+          {type: "v6e-8", cores: "8", runner: "linux-x86-ct6e-180-8tpu"},
+          {type: "v7x-8", cores: "8", runner: "linux-x86-tpu7x-224-4tpu"}
         ]
         python-version: ["3.11"]
         # Exclude v6e-8 tests for pypi_latest for resource constraints.

.github/workflows/wheel_tests_continuous.yml

@@ -241,7 +241,8 @@ jobs:
           tpu-specs: [
             # {type: "v3-8", cores: "4"}, # Enable when we have the v3 type available
             {type: "v5e-8", cores: "8", runner: "linux-x86-ct5lp-224-8tpu"},
-            {type: "v6e-8", cores: "8", runner: "linux-x86-ct6e-180-8tpu"}
+            {type: "v6e-8", cores: "8", runner: "linux-x86-ct6e-180-8tpu"},
+            {type: "v7x-8", cores: "8", runner: "linux-x86-tpu7x-224-4tpu"}
           ]
           libtpu-version-type: ["nightly"]
     name: "Pytest TPU (JAX artifacts version =  ${{ format('{0}', 'head') }})"

.github/workflows/wheel_tests_nightly_release.yml

@@ -177,7 +177,8 @@ jobs:
           tpu-specs: [
             # {type: "v3-8", cores: "4"}, # Enable when we have the v3 type available
             {type: "v5e-8", cores: "8", runner: "linux-x86-ct5lp-224-8tpu"},
-            {type: "v6e-8", cores: "8", runner: "linux-x86-ct6e-180-8tpu"}
+            {type: "v6e-8", cores: "8", runner: "linux-x86-ct6e-180-8tpu"},
+            {type: "v7x-8", cores: "8", runner: "linux-x86-tpu7x-224-4tpu"}
           ]
           libtpu-version-type: ["pypi_latest", "nightly"]
           exclude: