Enable elastic training for RL#3689

copybara-service[bot] wants to merge 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

test_901000964AI-Hypercomputer/maxtext:test_901000964Copy head branch name to clipboard