Books How to Scale Your Model Part 2 How to Think About TPUs How to Scale Your Model Part 3 Sharded Matrices and How to Multiply Them