教程

图片分类

• 在 Cloud TPU 上运行 MNIST (TF 2.x)

  一个使用 TensorFlow 的 MNIST 图片分类模型，针对在 Cloud TPU 上运行进行了优化。

• 在 Cloud TPU 上训练 ResNet (TF 2.x)

  一个使用 TensorFlow 的 ResNet 图片分类模型，针对在 Cloud TPU 上运行进行了优化。

• 在 Cloud TPU (TF 2.x) 上训练 Keras ResNet-RS

  一个使用 TensorFlow 的 Keras ResNet-RS 模型，针对在 Cloud TPU 上运行进行了优化。

• 在 Cloud TPU (PyTorch) 上训练 ResNet

  一个使用 PyTorch 的 ResNet 图片分类模型，针对在 Cloud TPU 上运行进行了优化。

• 在 Cloud TPU 上训练 EfficientNet (TF 2.x)

  一个使用 TensorFlow 的 EfficientNet 图片分类模型，针对在 Cloud TPU 上运行进行了优化。

对象检测

• 在 Cloud TPU 上训练 RetinaNet (TF 2.x)

  一个使用 TensorFlow 的 RetinaNet 对象检测模型，针对在 Cloud TPU 上运行进行了优化。

• 在 Cloud TPU 上训练 ShapeMask (TF 2.x)

  一个使用 TensorFlow 的 ShapeMask 对象检测模型，针对在 Cloud TPU 上运行进行了优化。

• 在 Cloud TPU 上训练 Mask RCNN (TF 2.x)

  一个使用 TensorFlow 的 Mask RCNN 模型，针对在 Cloud TPU 上运行进行了优化。

推荐系统

• 在 Cloud TPU 上训练 DLRM 和 DCN 模型 (TF 2.x)

  为点击率 (CTR) 预测等任务训练 DLRM 和 DCN v2 排名模型的指南。

• 在 Cloud TPU 上训练神经协同过滤模型 (TF 2.x)

  采用神经矩阵分解 (NeuMF) 模型的神经协同过滤 (NCF) 框架的实现

Pod 上的分布式处理

• 在 Cloud TPU Pod 上训练模型 (PyTorch)

  一个使用 PyTorch 的 ResNet-50 图片分类模型，针对在 Cloud TPU Pod 上运行进行了优化。

自然语言处理

• 使用 Cloud TPU 调整 BERT 模型：句子和句对分类任务 (TF 2.x)

  了解如何在 Cloud TPU 中使用基于 Transformer 的双向编码器表示法 (BERT)。

大语言模型

• 使用 Pax 在单设备 TPU 上训练

  使用 Pax 在单个设备上训练 SPMD 模型的指南 Cloud TPU。

• 在 v5e 上进行 JetStream MaxText 推理

  关于如何设置和使用 JetStream 与 MaxText 进行推理的指南。

• 在 v5e 上进行 JetStream PyTorch 推理

  有关如何设置和使用 JetStream 与 PyTorch 进行推理的指南。

• 使用 SAX 在 v5e 上提供 Llama 2

  使用 SAX 为 Llama 2 大语言模型系列提供服务的指南。