the path · 0/15
start the path

the whole stack · one map · four courses across it

One program, all the way down.

Two frameworks converge on one portable graph, cross the runtime seam under it, and come out as code for two different machines. Every chapter on this site is a floor or a seam of that descent. Walk far enough down and unfamiliar dumps stop being unfamiliar, which is the point: the courses end with kernels you derived, tested, and measured against the compiler.

no floor finished yet

Depth counts floors, not chapters. A floor lands when every unit you opened on it is finished, and 4/5 gates have passed on real hardware so far.

01 Read any IR on sight · narrate unfamiliar dumps, decode any dot_general
02 Write kernels that beat XLA · derived, differential-tested, benchmarked
03 Treat the cluster as one kernel · communication hidden behind compute
EX·03 streaming attention · a chapter 13 instrument, live
HBMQQKK0K1K2K3VV0V1V2V3VMEMqQ (resident)kAK0vAV0kBK1vBV1CARRIED STATEm : max(m, rowmax S1)l : l·e^(m₋−m) + rowsum e^(S1−m)acc : acc·e^(m₋−m) + e^(S1−m)·V1MXU · S = Q·K1ᵀ, then acc

03/06step 2 · one pass over K1, V1: rescale (m, l, acc), never materializing scores to HBM

scrub it: the carried (m, l, acc) state rescales exactly; the score matrix never exists

the map · each cell holds the units that live on that floor and the equipment that serves them · pick a course to light its route

the layers of the stack in order, then the stages that build on them enter the course →
the JAX language, from arrays to a training run enter the course →
the compiler at the waist, and the runtime seam above it enter the course →
eager PyTorch, then the bridges that land it on a TPU enter the course →
how deep you read
  • PyTorch side not yet
  • JAX side not yet
  • TPU side not yet
a floor counts once every unit you opened on it is finished, the waist included
  1. L0 the program you wrote
  2. L1 the trace
  3. L2 the portable graph
  4. the seam · PJRT · the program changes owners here
  5. L3 the compiler
  6. L4 the kernel
  7. L5 machine code
  8. L6 the silicon
  9. L7 the pod
under every cell

The workshop

Drills, reproduced failures and measured numbers all sit on the map now, under the floor they serve. These six instruments are the ones to open first, and ⌘K jumps to anything else.

the honesty rule

Gates pass only on real hardware, and failures stay published. Every value on this site states what it is: a roofline prediction, an off-chip verification, or a measured run with chip, dtype, shapes, and date attached.

measured on v6e-1 · the power rows · full provenance on the bench
pallas fused softmax, retuned222.7 µs vs 379.5 µs unfused, 267.6 µs XLA fused
windowed attention, mask as loop bound272.2 µs vs 1326.9 µs XLA masked (4.87x), 834.2 µs dense (3.06x)
the spill, byte-confirmed217.6 µs x 1287.5 GB/s = 280.2 MB vs 276.8 MB estimated: 1.2% agreement on a 20% bar