Add an example that train the torchtitan version of llama.

Add an example that train the torchtitan version of llama. #982