TPU

Creator
Creator
Seonglae ChoSeonglae Cho
Created
Created
2020 May 17 9:25
Editor
Edited
Edited
2026 Jul 21 16:46
Refs
Refs
An
ASIC
developed by Google, optimized for large-scale matrix multiplication processing and energy efficiency.
  • Systolic Array + Pipelining to minimize memory access
  • Ahead-of-Time compilation (XLA) to predetermine memory access patterns, utilizing scratchpads instead of caches
TPU Versions
 
 
 
https://jax-ml.github.io/scaling-book/
https://jax-ml.github.io/scaling-book/tpus/
https://jax-ml.github.io/scaling-book/tpus/
https://jax-ml.github.io/scaling-book/tpus/
Compute primitives and memory layouts for different hardware backends. Image by Chen et al., 2018.
Compute primitives and memory layouts for different hardware backends. Image by Chen et al., 2018.
https://jax-ml.github.io/scaling-book/tpus/
 
 
 
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.
github.com
Apple says its AI models were trained on Google's custom chips
Apple is using chips designed by Google in building its advanced AI models, according to a paper published on Monday.
Apple says its AI models were trained on Google's custom chips

Project Suncatcher

Expanding machine learning computation in space by building a satellite network equipped with TPUs that directly harness solar energy to perform large-scale AI computations
Meet Project Suncatcher, a research moonshot to scale machine learning compute in space.
Artificial intelligence is a foundational technology that could help us tackle humanity's greatest challenges. Now, we're asking where we can go next to unlock its fulle…
Meet Project Suncatcher, a research moonshot to scale machine learning compute in space.
What is a TPU?
Architecture, systolic arrays, and why Google's custom silicon still fights an uphill battle against CUDA
What is a TPU?
 
 
 

Recommendations