kd.typing.Any

class kauldron.typing.Any(*args, **kwargs)

Bases: object

Metadata describing a PyTree checkpoint.

A serialized PyTree structure with the same structure as the checkpointed PyTree. By “serialized”, we mean that the PyTree has been converted to a standardized representation, with all container nodes represented as standard types (e.g., tuple, list, dict, etc.). The leaves of the tree are individual parameter metadatas.