talk-data.com talk-data.com

Google Cloud Next session 2025-04-09 at 19:00

Fast train your LLMs by making TPUs go brrr

Topics

Description

Learn how to fast train your LLMs by maximizing utilization of TPUs. In this talk, we are going to start with training a GPT2 model with JAX from scratch on Colab and Kaggle TPUs for free. To scale it up, we will transition to the latest TPUs and the cutting edge MaxText library to train the state-of-the-art LLMs.