Publicaciones de Wei Wei

2 resultados

Borrar filtros
  • 12 DE AGOSTO DE 2025 / Kaggle

    Entrena un modelo GPT2 con JAX en una TPU gratis

    Crea y entrena un modelo GPT2 desde cero usando JAX en las TPU de Google, con un notebook completo de Python para Colab o Kaggle de nivel gratuito. Aprende a definir una malla de hardware, parámetros del modelo de partición y datos de entrada para el paralelismo de datos, y a optimizar el proceso de entrenamiento de modelos.

    Train a GPT2 model with JAX on TPU for free