Inference with the checkpoint

[mathmax12]

Though there are some related tickets, it would be great to see the steps to use the raw checkpoints for inference. https://github.com/pytorch/torchtitan/blob/main/docs/checkpoint.md This helps to verify the training works well.