Source: https://github.com/Lightning-AI/lit-gpt

Reproduction: https://github.com/amrayach/DS_PJ_4_LLM_DST

Dataset: https://huggingface.co/datasets/amay01/llm-sgd-dst8-train-test-data