Comment by jasonjmcghee

Highly recommend instead reading the human created "The Illustrated GPT-2" by Jay Alammar - https://jalammar.github.io/illustrated-gpt2/

And his similar work.

He also has a free course on "how llms work"