Add PyTorch checkpointing example to docs

[danieljanes]

Add an example for saving PyTorch checkpoints to the docs. The code is here:

https://github.com/adap/flower/discussions/911#discussioncomment-1743210

The relevant docs are here:

https://flower.dev/docs/saving-progress.html