Google I/O

ML frameworks for the gen AI era

Generative AI has captured everyone’s imagination. Come explore Google's best-in-class tools and frameworks for these use cases. We will cover JAX, a powerful framework for large-scale training and research that Google uses to build Gemini and Keras, a modeling framework for adapting gen AI OSS models that run on top of JAX, PyTorch, or TensorFlow.

Intermediate
Technical session
Join a community group

Meet developers, discover local groups, and build your global network.

Get started
Continue learning

Grow your skills around the Google technology you love.

Get started