TPU Deep Dive
I've been working with TPUs a lot recently and it's fun to see how they had such different design philosophies compared to GPUs. The main strongpoint for TPUs is in their scalability. This is achieved through a co-design of both the hardware side (e.g. energy efficiency and modularity) and the software side (e.g. XLA compiler). To give a brief tldr on TPUs, it's Google's ASIC that focuses on two factors: extreme matmul throughput + energy efficiency. Their origins go back to Google in 2006, when they were first evaluating whether they should implement either GPUs, FPGAs, or custom ASICs. Back then there were only a few applications that necessitated specialized hardware and they decided those needs could be met by bringing in excess CPU compute from their large datacenters. But this changed in 2013 when Google's voice search feature ran on neural networks and internal projections speculated that they would need much more compute if it took off. Fast forward to today and TPUs power the
TPU Deep Dive update: featured on Hacker News, thanks for reading! I've been working with TPUs a lot recently and it's fun to see how they had such different design philosophies compared to GPUs. The main strongpoint for TPUs is in their scalability. This is achieved through a co-design of both the hardware side (e.g. energy efficiency and modularity) and the software side (e.g. XLA compiler). Background To give a brief tldr on TPUs, it's Google's ASIC that focuses on two factors: extreme matmul throughput + energy efficiency. Their origins go back to Google in 2006, when they were first evalu
Explore this link on the map →saved by
related reading
- Touching the Elephant - TPUs | Consider the Bulldogconsiderthebulldog.com
- How To Scale Your Modeljax-ml.github.io
- How to Think About TPUs | How To Scale Your Modeljax-ml.github.io
- Tiny TPUtinytpu.com
- How to Think About GPUs | How To Scale Your Modeljax-ml.github.io
- Introduction to Cloud TPU | Google Cloud Documentationcloud.google.com
- The chip made for the AI inference era – the Google TPUuncoveralpha.com
- Understanding Matrix Multiplication on a Weight-Stationary Systolic Architecture | Telesenstelesens.co
- The Best GPUs for Deep Learning in 2023 — An In-depth Analysistimdettmers.com
- Inside NVIDIA GPUs: Anatomy of high performance matmul kernels - Aleksa Gordićaleksagordic.com
- TPU v4 enables performance, energy and CO2e efficiency gains | Google Cloud Blogcloud.google.com
- [1704.04760] In-Datacenter Performance Analysis of a Tensor Processing Unitarxiv.org