A Tensor Processing Unit, or TPU, is a custom AI chip designed by Google. Where a GPU started life as a graphics chip and was later adapted to AI, a TPU was built for neural networks from the start.
What a TPU does
Almost everything a neural network does comes down to multiplying large grids of numbers, called matrices, and adding up the results. A TPU devotes most of its silicon to exactly that job. Its core is a systolic array: a grid of small arithmetic units that pass data directly to their neighbors, so numbers flow through the chip without constantly being fetched from memory. Google's first TPU was described in a 2017 paper as running in its data centers since 2015.
Think of a kitchen built for one dish. A general-purpose chip is a full restaurant kitchen that can cook anything. A TPU is an assembly line that makes one thing very quickly and wastes little.
How it is used
TPUs are linked into large groups called pods, so thousands of chips can work on one model together. Google uses them to train and serve its own models, and rents them to cloud customers. Newer generations split into versions tuned for training and for inference, the two jobs that make up the cost of AI.
Why it matters to you
Most AI software is written against Nvidia's CUDA platform, which is a big reason Nvidia GPUs dominate. A TPU uses different software tools, so moving a workload between the two takes real engineering. That is why more places to rent TPUs matters: it gives builders a second supplier of compute, and a credible second supplier can put pressure on prices.
The tradeoff is portability. Code tuned tightly for one kind of chip may not run well on another. If you only call a model through an API, the chip underneath is the provider's problem. If you train or host your own models, it is yours.
Neighboring ideas
TPUs are one kind of AI accelerator. Others include Nvidia and AMD GPUs and custom chips built by cloud companies and labs. All of them depend on fast memory next to the processor, which is why high-bandwidth memory matters as much as the arithmetic units. The chip you pick affects speed, cost per token, and how large a model you can serve.
Common questions
Is a TPU better than a GPU? Neither wins everywhere. It depends on the model, the software and the price you are offered. Can you buy one? Historically TPUs were rented from Google Cloud, though Google has begun offering them in other ways. Check the provider's current terms.