From Jax to VLIW: Tracing a Computation Through the TPU Compiler Stack

patricktoulme.substack.com