Tutorials

Image Classification

• Training ResNet on Cloud TPU (PyTorch)

  A ResNet image classification model using PyTorch, optimized to run on Cloud TPU.

Distributed Processing on a Pod

• Training models on Cloud TPU Pods (PyTorch)

  A ResNet-50 image classification model using PyTorch, optimized to run on a Cloud TPU Pod.

Large Language Models

• JetStream MaxText inference on v6e

  A guide to set up and use JetStream with MaxText for inference on v6e.

• JetStream PyTorch inference on v6e

  A guide to set up and use JetStream with PyTorch for inference on v6e.

• MaxDiffusion inference on v6e

  A guide to set up and use MaxDiffusion for inference on v6e.

• vLLM inference on v6e

  A guide to set up and use vLLM for inference on v6e.

• Serve an LLM using TPUs on GKE with vLLM

  A guide to using vLLM to serve large language models (LLMs) using Tensor Processing Units (TPUs) on Google Kubernetes Engine (GKE).