Instructivos

Clasificación de imágenes

• Ejecuta MNIST en Cloud TPU (TF 2.x)

  Modelo de clasificación de imágenes MNIST que usa TensorFlow y está optimizado para ejecutarse en Cloud TPU.

• Entrena a un transformador en Cloud TPU (TF 2.x)

  Modelo de clasificación de imágenes ResNet que usa TensorFlow y está optimizado para ejecutarse en Cloud TPU.

• Entrena Keras ResNet-RS en Cloud TPU (TF 2.x)

  Un modelo Keras ResNet-RS que usa TensorFlow y está optimizado para ejecutarse Cloud TPU

• Entrena ResNet en Cloud TPU (PyTorch)

  Modelo de clasificación de imágenes ResNet que usa PyTorch y está optimizado para ejecutarse en Cloud TPU.

• Entrena EfficientNet en Cloud TPU (TF 2.x)

  Modelo de clasificación de imágenes EfficientNet que usa TensorFlow y está optimizado para ejecutarse en Cloud TPU.

Detección de objetos

• Entrena a un transformador en Cloud TPU (TF 2.x)

  Modelo de detección de objetos RetinaNet que usa TensorFlow y está optimizado para ejecutarse en Cloud TPU.

• Entrena ShapeMask en Cloud TPU (TF 2.x)

  Modelo de detección de objetos ShapeMask que usa TensorFlow y está optimizado para ejecutarse en Cloud TPU.

• Máscara de entrenamiento RCNN en Cloud TPU (TF 2.x)

  Modelo Mask RCNN que usa TensorFlow y está optimizado para ejecutarse en Cloud TPU.

Sistemas de recomendación

• Entrena modelos de DLRM y DCN en Cloud TPU (TF 2.x)

  Una guía para entrenar los modelos de clasificación DLRM y DCN v2 para tareas como la predicción de la tasa de clics (CTR).

• Entrena un modelo de filtrado colaborativo neuronal en Cloud TPU (TF 2.x)

  Una implementación del framework de filtrado colaborativo neuronal (NCF) con el modelo de factorización de matrices neuronal (NeuMF)

Procesamiento distribuido en un pod

• Entrenamiento de modelos en pods de Cloud TPU (PyTorch)

  Modelo de clasificación de imágenes ResNet-50 que usa PyTorch y está optimizado para ejecutarse en un pod de Cloud TPU.

Procesamiento de lenguaje natural

• BERT Función de ajuste con Cloud TPU: Tareas de clasificación de oraciones y pares (OF 2.x)

  Descubre cómo usar las representaciones de codificador bidireccional de transformadores (BERT) con Cloud TPU.

Modelos de lenguaje grandes

• Entrena en una TPU de un solo dispositivo con Pax

  Guía para entrenar un modelo de SPMD con Pax en un solo dispositivo Cloud TPU

• Inferencia de JetStream MaxText en v5e

  Una guía para configurar y usar JetStream con MaxText para la inferencia.

• Inferencia de PyTorch de JetStream en v5e

  Una guía para configurar y usar JetStream con PyTorch para la inferencia.

• Entrega Llama 2 en v5e con SAX

  Guía sobre el uso de SAX para entregar la familia de modelos grandes de lenguaje de Llama 2.