pytorch-lightning

9455871c - (2/n) Support 2D Parallelism - Distributed Checkpoints (#19852)

(2/n) Support 2D Parallelism - Distributed Checkpoints (#19852) * distributed checkpoints * use decorator * refactor if-strict * update example * filter non-persistent buffers (todo, add test) * simplify checkpoint loading for model