How to Scale Your Model – A Systems View of LLMs on TPUs (2025)
jax-ml.github.io