core.dist_checkpointing

Subpackages

• core.dist_checkpointing.strategies
  • core.dist_checkpointing.strategies.fully_parallel
  • core.dist_checkpointing.strategies.checkpointable
  • core.dist_checkpointing.strategies.zarr
  • core.dist_checkpointing.strategies.cached_metadata_filesystem_reader
  • core.dist_checkpointing.strategies.common
  • core.dist_checkpointing.strategies.torch
  • core.dist_checkpointing.strategies.async_utils
  • core.dist_checkpointing.strategies.filesystem_async
  • core.dist_checkpointing.strategies.tensorstore
  • core.dist_checkpointing.strategies.two_stage
  • core.dist_checkpointing.strategies.base
  • core.dist_checkpointing.strategies.state_dict_saver

Submodules

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