Add local_sa_* flags for local attention tuning#4125

aireenmei wants to merge 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

aireen/test_latest_changesAI-Hypercomputer/maxtext:aireen/test_latest_changesCopy head branch name to clipboard