I don’t think there is much more than just the doc: https://pytorch.org/docs/stable/checkpoint.html#torch.utils.checkpoint.checkpoint 23
It should be as simple as calling groups of layers from your network with the module.
You can find an advanced example in torchvision: https://github.com/pytorch/vision/blob/216035315185edec747dca8879d7197e7fb22c7d/torchvision/models/densenet.py 23