3 ms·
Everything thats in the blog post is basically well known already. Google publishes papers and gives talks about their TPUs. Many details are lacking though, an
by ipsum2 1y ago
Everything thats in the blog post is basically well known already. Google publishes papers and gives talks about their TPUs. Many details are lacking though, and require some assumptions/best guesses. Jax and XLA are (partially) open source and give clues about how TPUs work under the hood as well.
https://arxiv.org/abs/2304.01433 https://arxiv.org/abs/2304.01433
https://jax-ml.github.io/scaling-book/ https://jax-ml.github.io/scaling-book/