kd.ckpts.items

[[Source]]

Checkpoint handler.

Symbols

Class

kd.ckpts.items.CheckpointItem	Interface for a checkpoint item.
kd.ckpts.items.StandardCheckpointItem	Standard checkpoint item (for arbitrary jax.Array pytree).
kd.ckpts.items.TopLevelCheckpointItem	Checkpoint item that contains other sub-checkpoint items.