Supported models

Cloud TPU provides the following set of supported models that are optimized for fast and accurate training.

TensorFlow version Support start Support end Model category Supported models
TF 2.3 July 28, 2020 (End date not yet set) Image classification ResNet-2.x, MNIST-2.x, EfficientNet-2.x,
Language modeling Transformer-2.x, BERT-2.x
Object detection RetinaNet-2.x
Image segmentation Mask-RCNN-2.x, ShapeMask-2.x
Recommendation systems NCF-2.x
TF 2.2 May 8, 2020 (End date not yet set) Image classification ResNet-2.x, MNIST-2.x, EfficientNet-2.x,
Language modeling Transformer-2.x, BERT-2.x
Object detection RetinaNet-2.x
Image segmentation Mask-RCNN-2.x
TF 2.1.1 May 21, 2020 (End date not yet set) Image classification ResNet-2.x, MNIST-2.x
Language modeling Transformer-2.x, BERT-2.x
Object detection RetinaNet-2.x
TF 2.1 January 9, 2020 June 30, 2020 Image classification ResNet-2.x, MNIST-2.x
Language modeling Transformer-2.x, BERT-2.x
Object detection RetinaNet-2.x
TF 1.15.4 October 21, 2020 (End date not yet set) Image classification ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Language modeling Transformer, BERT
Object detection RetinaNet
Image segmentation Mask RCNN, ShapeMask
TF 1.15.3 May 29, 2020 (End date not yet set) Image classification ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Language modeling Transformer, BERT
Object detection RetinaNet
Image segmentation Mask RCNN, ShapeMask
TF 1.15.2 May 12, 2020 June 30, 2020 Image classification ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Language modeling Transformer, BERT
Object detection RetinaNet
Image segmentation Mask RCNN, ShapeMask
TF 1.15 October 22, 2019 June 30, 2020 Image classification ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
TF 1.14 July 19, 2019 June 30, 2020 Image classification ResNet, AmoebaNet,
EfficientNet,
MNasNet, MNIST
Language modeling
Object detection RetinaNet, Mask RCNN
TF 1.13 March 11, 2019 June 30, 2020 Image classification ResNet, AmoebaNet,
MNasNet, MNIST
Language modeling
Object detection RetinaNet
PyTorch version Support start Support end Model category Supported models
PyTorch 1.x Oct 17, 2019 (End date not yet set) Image classification ResNet
Language modeling FairSeq Transformer, FairSeq RoBERTa
Recommendation systems DLRM