Use custom maxtext branch

Rohan-Bierneni · Rohan-Bierneni · commit cd9daaf19d7c · 2025-09-26T18:59:16.000Z
diff --git a/.github/workflows/UnitTests.yml b/.github/workflows/UnitTests.yml
@@ -36,6 +36,7 @@ jobs:
         with:
           repository: AI-Hypercomputer/maxtext
           path: maxtext
+          ref: rbierneni-test-gpu-run
 
       - name: Run MaxText Training
         run: |
