This looks great for a first introduction to training LLMs, and it looks simple enough to try this locally. Great job!