deepspeedai / deepspeedai/DeepSpeed
Does the open-source code for FastPersist include the last two optimizations mentioned in the paper, "parallelizing checkpoint writes over DP ranks and pipelining checkpoint writes"?
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 43.1k
- Forks
- 5k
- Avg merge
- 4d 15h
- Merged PRs (30d)
- 112
Description
I followed the guide to perform checkpoint testing, and FastPersist performed very well. https://github.com/deepspeedai/DeepSpeedExamples/tree/master/deepnvme/model_checkpoint
I'd like to know if the open-source version of FastPersist includes the last two optimizations mentioned in the paper(https://arxiv.org/abs/2406.13768). If not, are there any plans to open-source these optimizations in the near future? Thank you.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the checkpoint testing guide in deepnvme/model_checkpoint and compare the open-source FastPersist implementation with the two optimizations described in the linked paper. Done means confirming whether parallelized checkpoint writes over DP ranks and pipelined writes are present, and documenting any stated plans to release them.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- distributed-systems, machine-learning, performance
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100