liveBook · Manning

TPU

As explained in Sutskever's List

A tensor processing unit is a specialized hardware accelerator designed to run and train artificial neural networks.

A tensor processing unit (TPU) is Google’s custom-built accelerator for running and training artificial neural networks. In GPipe experiments, TPU hardware provided the platform for scaling Transformer training across multiple partitions. section 6.3.2

Scaling a Transformer

A single 16 GB TPU can store approximately 282 million parameters. With GPipe, researchers trained an 83.9-billion-parameter Transformer across 128 TPU partitions—nearly 300 times the size achievable on one device. section 6.3.2

How Sutskever's List covers it

How deep it goes
Defines the TPU and gives per-device capacity, but does not describe its internal design.
Examples
Compares one 16 GB TPU’s capacity with an 83.9-billion-parameter Transformer trained across 128 partitions.
What you can do after
Understand how GPipe uses TPU partitions to train models far larger than a single device can store.
How it connects
Connects TPUs to GPipe, pipeline parallelism, model partitioning, rematerialization, and Transformer training.

Also explained in

Compare how 5 books explain TPU →

Related concepts

This explainer was generated by Manning's knowledge graph from the text of Sutskever's List. The section links point to the parts of the book it draws on.