Image Classification
-
Running MNIST on Cloud TPU (TF 2.x)
An MNIST image classification model using TensorFlow, optimized to run on Cloud TPU.
-
Training ResNet on Cloud TPU (TF 2.x)
A ResNet image classification model using TensorFlow, optimized to run on Cloud TPU.
-
Training Keras ResNet-RS on Cloud TPU (TF 2.x)
A Keras ResNet-RS model using TensorFlow, optimized to run on Cloud TPU.
-
Training ResNet on Cloud TPU (PyTorch)
A ResNet image classification model using PyTorch, optimized to run on Cloud TPU.
-
Training EfficientNet on Cloud TPU (TF 2.x)
An EfficientNet image classification model using TensorFlow, optimized to run on Cloud TPU.
Object Detection
-
Training RetinaNet on Cloud TPU (TF 2.x)
A RetinaNet object detection model using TensorFlow, optimized to run on Cloud TPU.
-
Training ShapeMask on Cloud TPU (TF 2.x)
A ShapeMask object detection model using TensorFlow, optimized to run on Cloud TPU.
-
Training Mask RCNN on Cloud TPU (TF 2.x)
A Mask RCNN model using TensorFlow, optimized to run on Cloud TPU.
Recommendation Systems
-
Training DLRM and DCN models on Cloud TPU (TF 2.x)
A guide to training DLRM and DCN v2 ranking models for tasks such as click-through rate (CTR) prediction.
-
Training a Neural Collaboration Filtering model on Cloud TPU (TF 2.x)
An implementation of the Neural Collaborative Filtering (NCF) framework with the Neural Matrix Factorization (NeuMF) model
Distributed Processing on a Pod
Natural Language Processing
Large Language Models
-
Train on a single-device TPU using Pax
A guide to train a SPMD model with Pax on a single-device Cloud TPU.
-
JetStream MaxText inference on v5e
A guide to set up and use JetStream with MaxText for inference.
-
JetStream PyTorch inference on v5e
A guide to set up and use JetStream with PyTorch for inference.
-
Serve Llama 2 on v5e using SAX
A guide to using SAX to serve the Llama 2 family of large language models.