[checkpoint] Expose thread count for checkpoint saving to control parallel writes per worker#2572

ananthsub merged 6 commits intoNVIDIA-NeMo:mainNVIDIA-NeMo/Megatron-Bridge:mainfrom

ananthsub:generalize-ckpt-assertionananthsub/Megatron-Bridge:generalize-ckpt-assertionCopy head branch name to clipboard