Official supported models

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

TensorFlow version Support start Support end Model category Supported models
TF 2.4.0 December 21, 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.3.1 October 1, 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.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.2 January 12, 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.2.1 October 7, 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.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.2 October 30, 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.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.5 January 14, 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.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