{"library":"orbax-checkpoint","install":[{"cmd":"pip install orbax-checkpoint jax flax","imports":["import orbax.checkpoint as ocp","from orbax.checkpoint import CheckpointManager","from orbax.checkpoint import CheckpointManagerOptions","from orbax.checkpoint import StandardSave","from orbax.checkpoint import StandardRestore"]},{"cmd":"pip install orbax-checkpoint","imports":[]}]}