Update patches 2026-01-12 + scaling + checkpoint restore#60

yeandy wants to merge 8 commits into

mainROCm/maxtext:mainfrom

yeandy/update-patches-2026-01-12-scaling-patch-checkpoint-restoreROCm/maxtext:yeandy/update-patches-2026-01-12-scaling-patch-checkpoint-restoreCopy head branch name to clipboard