This is by far the absolute best "how things go from a high-level (AI/LLM/Inference/Python/JAX/etc.) to low-level (TPU equivalent of assembly language instructions)" article that I've read to date!

Upvoted and favorited!

A keeper -- for present and future study...

Well done!