diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index eeacc5611e..b614e13d60 100644
--- a/.github/workflows/CI.yml
+++ b/.github/workflows/CI.yml
@@ -53,8 +53,6 @@ jobs:
           - macOS-latest
         test_group:
           - core
-          - neural_networks
-          - integration
         runtime:
           - "PJRT"
           - "IFRT"
@@ -66,21 +64,6 @@ jobs:
             assertions: false
             test_group: core
             runtime: "IFRT"
-          - os: ubuntu-24.04
-            version: "1.10"
-            assertions: true
-            test_group: core
-            runtime: "PJRT"
-          - os: ubuntu-24.04
-            version: "1.10"
-            assertions: true
-            test_group: neural_networks
-            runtime: "PJRT"
-          - os: ubuntu-24.04
-            version: "1.10"
-            assertions: true
-            test_group: integration
-            runtime: "PJRT"
           # - os: ubuntu-24.04
           #   libReactant: packaged
           #   version: '1.10'
@@ -93,6 +76,10 @@ jobs:
           #   libReactant: packaged
           #   version: '1.10'
           #   test_group: integration
+    env:
+      TPU_STDERR_LOG_LEVEL: 0
+      TF_CPP_MAX_VLOG_LEVEL: 3
+      TF_CPP_MIN_LOG_LEVEL: 0
     steps:
       - name: Set TMPDIR
         # We have to use `${GITHUB_WORKSPACE}` instead of `github.workspace` because GitHub
@@ -171,3 +158,7 @@ jobs:
       - uses: codecov/codecov-action@v5
         with:
           files: lcov.info
+      - uses: mxschmitt/action-tmate@v3
+        if: ${{ !cancelled() }}
+        with:
+          limit-access-to-actor: false