Use Tokamax's representative group sizes.#3434

copybara-service[bot] merged 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

test_885246948AI-Hypercomputer/maxtext:test_885246948Copy head branch name to clipboard