TPU software versions

The software version to pass when you create a TPU VM, by generation and framework.

Versions by generation

Generation	PyTorch and JAX	TensorFlow
TPU v5e	v2-alpha-tpuv5-lite	Version pattern tpu-vm-tf-x.y.z-{pod}-pjrt, TensorFlow 2.15.0 or later
TPU v5p	v2-alpha-tpuv5	Version pattern tpu-vm-tf-x.y.z-{pod}-pjrt, TensorFlow 2.15.0 or later
TPU v6e (Trillium)	v2-alpha-tpuv6e	Version pattern tpu-vm-tf-x.y.z-{pod}-pjrt, TensorFlow 2.15.0 or later
TPU7x (Ironwood)	None listed	Not supported

For TensorFlow, include pod in the version for multi-host slices and omit it for single-host TPUs, for example tpu-vm-tf-2.16.0-pod-pjrt.

Important: The Cloud TPU API is no longer under active development and is not supported on TPU7x (Ironwood) and later. Create TPU7x (Ironwood) capacity with Google Kubernetes Engine (GKE) or Compute Engine.

Related information

Networking and access for Cloud TPU VMs

Create a TPU VM