07:00
2026-07-27
hiraditya.github.io
machine-learning
Programming the TPU: What Its Open-Source Compiler Already Tells You
Google's open-source TPU compiler, shipped as the Mosaic TPU dialect inside JAX, documents eight explicit memory spaces that programmers must manage, including vector memory (VMEM), scalar memory (SME…