Skip to main content
Log in

Mojo package

checkpoint

APIs to save and load checkpoints for MAX graphs.

Modules

  • metadata: Tools for reading metadata from a checkpoint file without completely loading all tensors.
  • save_load: Defines functions to save/load tensors from a checkpoint file.
  • tensor_dict: Structs used for maintaining a collection of tensors.