CI/CD UT optimization#3753

charlesli640 wants to merge 2 commits intomainAI-Hypercomputer/maxtext:mainfrom

charlesli/ut_bestpracticeAI-Hypercomputer/maxtext:charlesli/ut_bestpracticeCopy head branch name to clipboard