I am referring the paper “On Efficient Constructions of Checkpoints” https://proceedings.icml.cc/static/paper_files/icml/2020/5071-Paper.pdf 2
I am not sure this is implemented or not, my implementation is based on this paper.
If code for this research project is implemented, could you please point me or share the link.