Modelos de referência compatíveis

O Cloud TPU fornece um conjunto de modelos de referência otimizados para treinamento rápido e preciso.

O Cloud TPU oferece suporte às seguintes versões principais e secundárias do framework do TensorFlow, PyTorch e JAX/FLAX.

A numeração das versões do TensorFlow mudou com a versão 2.5.0. Os números principais de versão do TensorFlow terminam com "0", e todos os números de versão de patch terminam com números maiores que "0". Por exemplo, TF 2.10.0 é uma versão principal e TF 2.10.1 é uma versão secundária.

Para executar a versão mais recente do framework, verifique se há alguma versão de patch para a versão principal. Se houver, execute a versão de patch compatível mais recente em vez da versão principal.

Framework	Versão principal	Categoria do modelo	Modelos de referência	Versões compatíveis
PyTorch	2.x	Classificação de imagens	ResNet-PyTorch	1.13, 2.0
		JAX	mais recente	Modelos de linguagem grande	MaxText	mais recente
JAX/FLAX	mais recente	Classificação de imagens	ResNet50	mais recente