Skip to main content

Technology

TPUs

Tensor Processing Units designed specifically for machine learning matrix operations and neural network training.

5 companies

Explainer

TPUs explained

A TPU, or tensor processing unit, is a chip designed by Google to run machine learning. A tensor is a grid of numbers, and neural networks are mostly built from multiplying such grids. A TPU is organised around doing that one operation very efficiently.

The core of a TPU is a systolic array, a grid of multiply-accumulate units where data flows from one unit to the next in a regular rhythm. This reuses each value many times without going back to memory. Google began using TPUs internally in 2015 and now offers them to customers through Google Cloud.

How it differs from the usual approach

A GPU is a general parallel processor that has been extended for AI. A TPU was designed from the start for neural network maths, and it gives up some flexibility for efficiency on those workloads. TPUs are mainly rented through Google Cloud rather than bought as chips, though Google has said it will supply a few large customers directly, and they are programmed through frameworks such as JAX and TensorFlow, with PyTorch support also available.

Where it is used

  • Training and serving Google's own models and products, such as search and translation.
  • Renting TPU capacity on Google Cloud for model training and inference.
  • Large-scale training jobs spread across many TPUs joined in pods.
  • Research workloads using the JAX framework.
  • Third-party AI developers who choose TPUs as an alternative to GPUs.

What to watch

  • Google publishes new TPU generations on a regular cycle, with later ones focused on inference as well as training.
  • Other companies use "TPU-like" designs, so the term is sometimes applied to any matrix-focused AI chip.
  • Demand for non-NVIDIA capacity makes TPUs a visible alternative for large AI developers.
  • Software support and portability remain the practical question for teams moving between TPUs and GPUs.

Frequently asked questions

What does TPU stand for?

TPU stands for tensor processing unit. It is a chip family designed by Google for machine learning workloads.

Can I buy a TPU?

Mostly no. Google uses TPUs in its own data centers and rents them through Google Cloud. It has said it will also sell them to a small number of large customers, but they are not sold as general-purpose chips the way GPUs are.

Is a TPU better than a GPU?

It depends on the workload and software. TPUs can be efficient for large matrix-heavy models. GPUs are more flexible and have wider software support.