Make the latest checkpoint step consistent across all evaluation processes.
Make the latest checkpoint step consistent across all evaluation processes.

PiperOrigin-RevId: 470854098

google/paxml,google/paxml