Add checkpoint resharding script for faster loading#3801

shuningjin wants to merge 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

shuningjin-reshardAI-Hypercomputer/maxtext:shuningjin-reshardCopy head branch name to clipboard