Add support for partial checkpoint restore#1357

terrykong wants to merge 1 commit into

google-research:maingoogle-research/t5x:mainfrom

terrykong:patch/partial-checkpoint-restoreterrykong/t5x:patch/partial-checkpoint-restoreCopy head branch name to clipboard