LICENSE
README.md
pyproject.toml
tests/test_barriers.py
tests/test_builder.py
tests/test_checkpoint_base.py
tests/test_checkpoint_loader.py
tests/test_checkpoint_manager.py
tests/test_checkpoint_process.py
tests/test_checkpoint_reader.py
tests/test_checkpoint_reader_mmap.py
tests/test_checkpoint_saver.py
tests/test_checkpoint_writer.py
tests/test_checkpointer.py
tests/test_consolidation.py
tests/test_distributed_metadata.py
tests/test_dtensor_resharder.py
tests/test_dtensor_resharder_unit.py
tests/test_local_filesystem.py
tests/test_lock.py
tests/test_logging_utils.py
tests/test_metadata_manager.py
tests/test_safetensors_serialization.py
tests/test_schema.py
tests/test_staging.py
tests/test_state_transformations.py
tests/test_torch_serialization.py
tests/test_types.py
tests/test_utils.py
tests/test_version.py
tests/test_walk.py
torch_checkpointing/__init__.py
torch_checkpointing/_pin_memory_utils.py
torch_checkpointing/_state_dict_stager.py
torch_checkpointing/barriers.py
torch_checkpointing/builder.py
torch_checkpointing/checkpoint_base.py
torch_checkpointing/checkpoint_layout.py
torch_checkpointing/checkpoint_loader.py
torch_checkpointing/checkpoint_manager.py
torch_checkpointing/checkpoint_process.py
torch_checkpointing/checkpoint_reader.py
torch_checkpointing/checkpoint_saver.py
torch_checkpointing/checkpoint_writer.py
torch_checkpointing/checkpointable.py
torch_checkpointing/config.py
torch_checkpointing/distributed_metadata.py
torch_checkpointing/dtensor_metadata.py
torch_checkpointing/dtensor_resharder.py
torch_checkpointing/lock.py
torch_checkpointing/logging_utils.py
torch_checkpointing/metadata_manager.py
torch_checkpointing/py.typed
torch_checkpointing/resharding.py
torch_checkpointing/resharding_utils.py
torch_checkpointing/schema.py
torch_checkpointing/staging.py
torch_checkpointing/state_transformations.py
torch_checkpointing/types.py
torch_checkpointing/utils.py
torch_checkpointing/version.py
torch_checkpointing/walk_utils.py
torch_checkpointing.egg-info/PKG-INFO
torch_checkpointing.egg-info/SOURCES.txt
torch_checkpointing.egg-info/dependency_links.txt
torch_checkpointing.egg-info/requires.txt
torch_checkpointing.egg-info/top_level.txt
torch_checkpointing/hf/__init__.py
torch_checkpointing/hf/consolidation.py
torch_checkpointing/storage/base_storage.py
torch_checkpointing/storage/filesystem.py
torch_checkpointing/storage/torch_serialization.py