Fix eval and checkpointing under load-balanced CP#4613

huytransformer wants to merge 1 commit into

AI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

huytransformer:htn/cp-reorder-iteratorhuytransformer/maxtext:htn/cp-reorder-iteratorCopy head branch name to clipboard