.gitignore
.pre-commit-config.yaml
LICENCE
README.md
pyproject.toml
.github/workflows/cpu-tests.yml
.github/workflows/pre-commit.yaml
.github/workflows/python-publish.yml
checkpoint_engine/__init__.py
checkpoint_engine/__main__.py
checkpoint_engine/_version.py
checkpoint_engine/api.py
checkpoint_engine/data_types.py
checkpoint_engine/device_utils.py
checkpoint_engine/p2p_store.py
checkpoint_engine/pin_memory.py
checkpoint_engine/ps.py
checkpoint_engine/worker.py
checkpoint_engine.egg-info/PKG-INFO
checkpoint_engine.egg-info/SOURCES.txt
checkpoint_engine.egg-info/dependency_links.txt
checkpoint_engine.egg-info/requires.txt
checkpoint_engine.egg-info/top_level.txt
checkpoint_engine/distributed/__init__.py
checkpoint_engine/distributed/base.py
checkpoint_engine/distributed/vllm_hccl.py
checkpoint_engine/distributed/vllm_nccl.py
docs/npu_start.md
examples/update.py
figures/checkpoint-engine.png
figures/overlap-update-and-copy.png
figures/pipeline.png
patches/vllm_fp8.patch
tests/test_assign_receiver_ranks.py
tests/test_inplace_unpin.py
tests/test_rdma_parser.py
tests/test_reuse_pin_memory.py
tests/test_update.py