Write For Us

Optimizing Flax NNX Models with Optax (Part 2)

E-Commerce Solutions SEO Solutions Marketing Solutions
10 Views
Published
This is part two of our two episode series on optimizing models with Optax! This episode builds on the foundation of the first, exploring the powerful, composable features of Optax and showing how to scale training to multiple devices.

Resources:
Learn more → https://goo.gle/learning-jax

Subscribe to Google for Developers → https://goo.gle/developers

Speaker: Robert Crowe
Category
Project
Tags
Google, developers, pr_pr: AI DevRel (fka Core ML);
Sign in or sign up to post comments.
Be the first to comment