AI, Data Science & Business Certificates from Google, IBM & Microsoft
Build with Azure OpenAI, Copilot Studio & Agentic Frameworks — Microsoft Certified
Overview
Google, IBM & Meta Certificates — All 10,000+ Courses at 40% Off
One annual plan covers every course and certificate on Coursera. 40% off for a limited time.
Get Full Access
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