maxtext icon indicating copy to clipboard operation
maxtext copied to clipboard

Published 20 hours ago • verified publisher icon google

add compatibility w orbax's new API and add an opttion to restore wit…

Open ssusie opened this issue 11 months ago • 0 comments

Changing checkpointing to use the new API and replacing default with items. Also added an option to restore checkpoints with SingleReplicaArrayHandler

ssusie avatar Mar 08 '24 18:03 ssusie