Skip to content

References: How models run on hardware

Source curriculum (structural mirror, cited as further study):
• Stanford CS336, "Language Modeling from Scratch", Lecture 5: GPUs, TPUs
Instructors: Tatsunori Hashimoto and Percy Liang (Stanford)
Course page: https://cs336.stanford.edu/
Lecture videos: YouTube playlist
https://www.youtube.com/playlist?list=PLoROMvodv4rMqXOcazWaTUHhq-yembLCV
License: no explicit license is published on the course site; lecture
videos are on YouTube under standard terms; slides are public on GitHub
without a stated license.
Required attribution: "Based on the structure of Stanford CS336,
'Language Modeling from Scratch,' by Tatsunori Hashimoto and Percy Liang
(cs336.stanford.edu). This is an independent structural mirror in
original prose; it reproduces no course materials, and Stanford does
not endorse it."
This lesson mirrors the structure of Lecture 5 (GPU/TPU execution model and
memory hierarchy). Clawdemy's lessons are original prose that follows the
pedagogical arc of the course. Because the source publishes no explicit
license, we cite it as a recommended companion and reproduce none of its
materials. All rights to the original course materials remain with their
creators.
  • Stanford CS336, Lecture 5: GPUs and TPUs by Hashimoto and Liang. The lecture this lesson mirrors. It goes further into bandwidth numbers and the precise mapping of compute units, useful when you start writing code that has to hit those tiles.

A short, durable list. Each link is a specific next step, not a generic pile.

Where this connects inside the track.

  • Counting the cost (lesson 2). Arithmetic intensity from there is the lens this lesson uses physically: the gap between HBM and the compute units is what makes the number matter.

  • Writing fast kernels: Triton and XLA (lesson 6). The next lesson turns “stage data into fast memory and reuse it” into code, with custom kernels that fuse operations to raise intensity. The FlashAttention idea.

  • Parallelism (lesson 7). Once a single GPU is fully used, the next move is many of them; that lesson uses this hardware picture to explain why each parallelism scheme exists.