checkpoint utility: optimize to_maxtext, add deepseek#3184

copybara-service[bot] merged 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

shuningjin-ckpt-opt3AI-Hypercomputer/maxtext:shuningjin-ckpt-opt3Copy head branch name to clipboard