talk-data.com
Google Cloud Next
session
2025-04-09 at 19:00
Fast train your LLMs by making TPUs go brrr
Event:
Google Cloud Next '25
Speakers
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.