Module checkpointing 

State checkpointing for parameter management

Modules

utils

Utility functions for checkpointing

Structs

CheckpointManager

In-memory checkpoint manager

CheckpointMetadata

Metadata for checkpoints

OptimizerCheckpoint

Checkpoint data for optimizer state

ParameterGroupCheckpoint

Checkpoint data for a parameter group

Traits

Checkpointable

Trait for optimizers that support checkpointing