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.

Follow the instructions in Set up the Cloud TPU environment to prepare to create a Cloud TPU.

The VM and Pod quickstarts provide a brief introduction to working with Cloud TPU VMs using TensorFlow, JAX, and PyTorch. These topics explain how to install an ML framework and run a sample application on a Cloud TPU VM.

Set up your environment

• Set up the Cloud TPU environment

  Configure your Google Cloud account and project for using Cloud TPUs, by activating the Cloud TPU API, setting up permissions, and installing tools.

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.