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	Transformer
			Object detection	RetinaNet, Mask RCNN
TF 1.13	March 11, 2019	June 30, 2020	Image classification	ResNet, AmoebaNet, MNasNet, MNIST
			Language modeling	Transformer
			Object detection	RetinaNet

The tf-nightly version of TensorFlow is available for experimental use only. It is built nightly and may be unstable. It is not officially supported and should not be used in production environments.

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