How to Think About TPUs
https://jax-ml.github.io/scaling-book/tpus Part 2 of How To Scale Your Model (Part 1: Rooflines | Part 3: Sharding) This section is all about how TPUs work, how they're networked together to enable multi-chip training and inference,…