Official supported models

Cloud TPU with TensorFlow provides the following set of official, supported reference models that have been optimized for fast and accurate training.

Framework Major version Model category Reference models Supported versions
TensorFlow 2.x Image classification ResNet-2.x, MNIST-2.x, EfficientNet-2.x 2.1, 2.2, 2.3, 2.4
Language modeling Transformer-2.x, BERT-2.x 2.1, 2.2, 2.3, 2.4
Language modeling XlNet-2.x 2.4
Object detection RetinaNet-2.x 2.1, 2.2, 2.3, 2.4
Image segmentation Mask-RCNN-2.x 2.2, 2.3, 2.4
Image segmentation ShapeMask-2.x 2.3, 2.4
Recommendation systems NCF-2.x 2.3, 2.4
1.x Image classification ResNet, AmoebaNet, MNasNet, MNIST, EfficientNet 1.15
Language modeling Transformer, BERT 1.15
Object detection RetinaNet 1.15
PyTorch 1.x Image classification ResNet-PyTorch 1.7
Language modeling FairSeq Transformer, FairSeq RoBERTa 1.7
Recommendation systems DLRM 1.7