支持的参考模型
Cloud TPU 提供了一组参考模型，这些模型经过优化，可实现快速准确的训练。
Cloud TPU 支持 TensorFlow、PyTorch 和 JAX/FLAX 的以下主要和次要框架版本。
在 2.5.0 版中，TensorFlow 版本编号已更改。TensorFlow 主要版本号以“0”结尾，所有补丁版本号均以大于“0”的数字结尾。例如，TF 2.10.0 是主要版本，TF 2.10.1 是次要版本。
如需运行最新的受支持框架版本，请检查主要版本是否有任何补丁版本。如果是这样，您可以运行最新的受支持补丁版本，而不是主要版本。
| 框架 | 主要版本 | 模型类别 | 参考模型 | 支持的版本 |
|---|---|---|---|---|
| TensorFlow | 2.x | 图片分类 | ResNet-2.x、MNIST-2.x、EfficientNet-2.x | 请参阅支持的 TensorFlow 版本。 |
| 语言建模 | Transformer-2.x、BERT-2.x | 请参阅支持的 TensorFlow 版本。 | ||
| 对象检测 | RetinaNet-2.x | 请参阅支持的 TensorFlow 版本。 | ||
| 图片分割 | Mask-RCNN-2.x、ShapeMask-2.x | 请参阅支持的 TensorFlow 版本。 | ||
| 推荐系统 | DLRM-2.x、DCN-2.x、NCF-2.x | 请参阅支持的 TensorFlow 版本。 | ||
| PyTorch | 2.x | 图片分类 | ResNet-PyTorch | 1.13、2.0 |
| 图片生成 | Stable Diffusion、 | 1.13、2.0 | ||
| JAX | 最新 | 大型语言模型 | MaxText | latest |
| JAX/FLAX | 最新 | 图片分类 | ResNet50 | latest |