Integrate Orbax's emergency checkpoint.#820

Draft

hanzhi713 wants to merge 1 commit intoapple:mainfrom hanzhi713:in-mem-ckpt

+1,085-5

Commits on Jan 8, 2025

Integrate Orbax emergency checkpoint
hanzhi713
committed