Search

2 results

Clear filters
  • AUG. 12, 2025 / Kaggle

    Train a GPT2 model with JAX on TPU for free

    Build and train a GPT2 model from scratch using JAX on Google TPUs, with a complete Python notebook for free-tier Colab or Kaggle. Learn how to define a hardware mesh, partition model parameters and input data for data parallelism, and optimize the model training process.

    Train a GPT2 model with JAX on TPU for free
  • OCT. 22, 2024 / Gemma

    Introducing Keras Hub: Your one-stop shop for pretrained models

    KerasHub is a new unified library for pretrained models fostering a more cohesive ecosystem for developers.

    Keras on Google for Developers