Let's build GPT from scratch w/ Nx and Axon

[theodore]

Thanks Nick Yea it was quite a grind to figure it out. I actually used some of your notebooks to learn Nx + Axon, so thank you too!