Class Central is learner-supported. When you buy through links on our site, we may earn an affiliate commission.

Google

Train Your JAX Models Using model.fit(...) in Keras 3

Google via YouTube

Start learning Write review
Corporate Finance Institute Ad

35% Off Finance Skills That Get You Hired - Code CFI35

Learn More → Datacamp Ad

Power BI Fundamentals - Create visualizations and dashboards from scratch

Learn More →

Overview

Coursera Flash Sale
40% Off Coursera Plus for 3 Months!
Grab it
This 20-minute Google I/O 2025 talk explores how to leverage Keras 3 for JAX model training using the familiar model.fit(...) API pattern. Discover multiple entry points into the JAX ecosystem through Keras, including KerasHub's catalog of over 100 model architectures with support for thousands of HuggingFace checkpoints. Learn how to fine-tune and optimize models using techniques like LoRA and qLoRA with JAX or other backends of your choice. The speakers, Divyashree Sreepathihalli, Monica Song, and Fabien Hertschuh, also introduce Keras Recommenders, the newest addition to the Keras ecosystem. Access comprehensive resources including the Keras documentation, Keras Hub, Keras Recommenders, Keras on Kaggle, Keras on Hugging Face, and developer guides to enhance your JAX model training experience.

Syllabus

Train your JAX models using model.fit(...) in Keras 3

Taught by

Google Developers

Tags

Reviews

Start your review of Train Your JAX Models Using model.fit(...) in Keras 3

Start learning

Never Stop Learning.

Get personalized course recommendations, track subjects and courses with reminders, and more.

Sign up for free
Someone learning on their laptop while sitting on the floor.