Get started

Before starting one of the Cloud TPU VM quickstarts, read Introduction to Cloud TPU which gives an overview of working with Cloud TPUs.

These quickstarts provides a brief introduction to working with Cloud TPU VMs using TensorFlow, JAX, and PyTorch. Each of the following topics explains how to install an ML framework and run a sample application on a Cloud TPU VM.

Cloud TPU VM quickstarts

• Run TensorFlow on Cloud TPU VMs

  Run a simple calculation on Cloud TPU VM with TensorFlow.

• Run JAX on Cloud TPU VMs

  Run a simple calculation on Cloud TPU VM with JAX.

• Run PyTorch on Cloud TPU VMs

  Run a simple calculation on Cloud TPU VM with PyTorch.

Cloud TPU Pod quickstarts

• Train on Cloud TPU Pods

  Learn about best practices and common issues for Cloud TPU Pods.

• Run TensorFlow on Cloud TPU Pod slices

  Run a simple calculation on Cloud TPU Pod slice with TensorFlow.

• Run JAX on Cloud TPU Pod slices

  Run a simple calculation on Cloud TPU Pod slice with JAX.

• Run PyTorch on Cloud TPU Pod slices

  Run a simple calculation on Cloud TPU Pod slice with PyTorch.