In that case, you might also enjoy https://jaykmody.com/blog/gpt-from-scratch/ https://jaykmody.com/blog/gpt-from-scratch/ (here's the raw code: https://github

In that case, you might also enjoy https://jaykmody.com/blog/gpt-from-scratch/ https://jaykmody.com/blog/gpt-from-scratch/ (here's the raw code: https://github.com/jaymody/picoGPT/blob/main/gpt2.py https://github.com/jaymody/picoGPT/blob/main/gpt2.py)