core.dist_checkpointing

Subpackages

• core.dist_checkpointing.strategies
  • core.dist_checkpointing.strategies.common
  • core.dist_checkpointing.strategies.checkpointable
  • core.dist_checkpointing.strategies.torch
  • core.dist_checkpointing.strategies.fully_parallel
  • core.dist_checkpointing.strategies.nvrx

Submodules

• core.dist_checkpointing.optimizer
• core.dist_checkpointing.dict_utils
• core.dist_checkpointing.validation
• core.dist_checkpointing.tensor_aware_state_dict
• core.dist_checkpointing.gpt_checkpoint_interop
• core.dist_checkpointing.serialization
• core.dist_checkpointing.utils
• core.dist_checkpointing.core
• core.dist_checkpointing.exchange_utils
• core.dist_checkpointing.mapping
• core.dist_checkpointing.state_dict_utils