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
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.