[ckpt] fix: Prevent int32 overflow for high train sample counts.#3290

BlueCrescent wants to merge 1 commit intoNVIDIA-NeMo:mainNVIDIA-NeMo/Megatron-Bridge:mainfrom

BlueCrescent:fix_sample_count_type_in_checkpoint_stateBlueCrescent/Megatron-Bridge:fix_sample_count_type_in_checkpoint_stateCopy head branch name to clipboard