Comment Comment by desideratum The Scaling ML textbook also has an excellent section on TPUs. https://jax-ml.github.io/scaling-book/tpus/ parent desideratum 11 months ago