LICENSE
README.md
pyproject.toml
setup.py
tests/test_async_take.py
tests/test_batcher.py
tests/test_chunked_tensor_io_preparer.py
tests/test_ddp.py
tests/test_ddp_infer_replication.py
tests/test_ddp_replication_glob.py
tests/test_dist_store.py
tests/test_flatten.py
tests/test_fs_storage_plugin.py
tests/test_gcs_storage_plugin.py
tests/test_manifest.py
tests/test_memoryview_stream.py
tests/test_partitioner.py
tests/test_pg_wrapper.py
tests/test_read_object.py
tests/test_replication_glob.py
tests/test_rng_state.py
tests/test_rss_profiler.py
tests/test_s3_storage_plugin.py
tests/test_serialization.py
tests/test_sharded_tensor_io_preparer.py
tests/test_sharded_tensor_resharding.py
tests/test_snapshot.py
tests/test_state_dict.py
tests/test_tensor_io_preparer.py
tests/test_test_utils.py
tests/test_uvm_tensor.py
torchsnapshot/__init__.py
torchsnapshot/batcher.py
torchsnapshot/dist_store.py
torchsnapshot/dtensor_utils.py
torchsnapshot/event.py
torchsnapshot/event_handlers.py
torchsnapshot/flatten.py
torchsnapshot/io_preparer.py
torchsnapshot/io_types.py
torchsnapshot/knobs.py
torchsnapshot/manifest.py
torchsnapshot/manifest_ops.py
torchsnapshot/manifest_utils.py
torchsnapshot/memoryview_stream.py
torchsnapshot/partitioner.py
torchsnapshot/pg_wrapper.py
torchsnapshot/py.typed
torchsnapshot/rng_state.py
torchsnapshot/rss_profiler.py
torchsnapshot/scheduler.py
torchsnapshot/serialization.py
torchsnapshot/snapshot.py
torchsnapshot/state_dict.py
torchsnapshot/stateful.py
torchsnapshot/storage_plugin.py
torchsnapshot/test_utils.py
torchsnapshot/uvm_tensor.py
torchsnapshot/version.py
torchsnapshot/io_preparers/__init__.py
torchsnapshot/io_preparers/chunked_tensor.py
torchsnapshot/io_preparers/dtensor.py
torchsnapshot/io_preparers/object.py
torchsnapshot/io_preparers/sharded_tensor.py
torchsnapshot/io_preparers/tensor.py
torchsnapshot/storage_plugins/__init__.py
torchsnapshot/storage_plugins/fs.py
torchsnapshot/storage_plugins/gcs.py
torchsnapshot/storage_plugins/s3.py
torchsnapshot/tricks/__init__.py
torchsnapshot/tricks/ddp.py
torchsnapshot/tricks/deepspeed.py
torchsnapshot/tricks/fsdp.py
torchsnapshot_nightly.egg-info/PKG-INFO
torchsnapshot_nightly.egg-info/SOURCES.txt
torchsnapshot_nightly.egg-info/dependency_links.txt
torchsnapshot_nightly.egg-info/requires.txt
torchsnapshot_nightly.egg-info/top_level.txt
torchsnapshot_nightly.egg-info/zip-safe