Optimizing Flax NNX Models with Optax (Part 1)
219
Google Developers2.61 млн
Следующее
93 дня – 2947:13Популярные
33 дня – 6 2000:32
51 день – 5 5250:50Опубликовано 4 декабря 2025, 5:00
This is the first of our two part series on optimizing models with Optax. Let’s explore how to optimize Flax NNX neural network models using Optax, the primary optimization library in the JAX ecosystem. This presentation is designed especially for those of you familiar with PyTorch, and we'll be drawing parallels and highlighting differences to help you transition smoothly. This first episode will cover the core workflow, rom defining a model to running a complete, JIT-compiled training loop. It establishes the foundational knowledge that we’ll need for the more advanced topics in the next episode.
Resources:
Learn more → goo.gle/learning-jax
Subscribe to Google for Developers → goo.gle/developers
Speaker: Robert Crowe
Resources:
Learn more → goo.gle/learning-jax
Subscribe to Google for Developers → goo.gle/developers
Speaker: Robert Crowe
Свежие видео
4 дня – 16 97515:59
7 дней – 58 8460:31
8 дней – 615 51610:00
8 дней – 10:02
9 дней – 215 8822:08
17 дней – 34 78112:39Случайные видео
189 дней – 1 718 7310:25
221 день – 33 25223:27
11.10.24 – 4690:39
03.03.23 – 6523:32
1 день – 12 0310:15
1 день – 87 00222:07
3 дня – 9 9531:18
8 дней – 9 6793:59
10 дней – 1310:40
