pytorch

7a772bff - [dtensor] add submesh example to checkpoint_example (#95655)

[dtensor] add submesh example to checkpoint_example (#95655) This PR adds a submesh example for checkpoing purposes Pull Request resolved: https://github.com/pytorch/pytorch/pull/95655 Approved by: https://github.com/XilunWu