Cloud TPU 教程

图片分类

在 Cloud TPU 上运行 MNIST (TF 2.x)
在 Cloud TPU 上训练 ResNet (TF 2.x)
在 Cloud TPU 上训练 (TF 2.x)
使用 PyTorch 在 Cloud TPU 上训练 Resnet50
在 Cloud TPU 上训练 EfficientNet (TF 2.x)

对象检测

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

推荐系统

在 Cloud TPU 上训练 DLRM 和 DCN (TF 2.x)
在 Cloud TPU 上训练 NCF (TF 2.x)

Pod上的分布式处理

在 Cloud TPU Pod 上训练 PyTorch 模型

自然语言处理

使用 Cloud TPU 进行 BERT 微调：句子和句对分类任务 (TF 2.x)

大语言模型

使用 Pax 在单主机 TPU 上训练
在 v5e Cloud TPU 虚拟机上进行 JetStream MaxText 推理
在 v5e Cloud TPU 虚拟机上进行 JetStream PyTorch 推断
使用 SAX 在 TPU v5e 上部署和提供 Llama 2 模型

文章来源：google cloud

推荐阅读

• 使用 SAX 在 TPU v5e 上部署和提供 Llama 2 模型
• 在 v5e Cloud TPU 虚拟机上进行 JetStream PyTorch 推断
• 在 v5e Cloud TPU 虚拟机上进行 JetStream MaxText 推理

更多芯擎AI开发板干货请关注芯擎AI开发板专栏。欢迎添加极术小姐姐微信（id:aijishu20)加入技术交流群，请备注研究方向。