Pytorch Example Code For Llm