[Example]Expand the nanogpt example to babyllama.

[Antlera]

Llama is the most popular open-source LLM base model. Try to support llama training or fine-tuning. BabyLlama should be a good startup example.