Debugging JAX & Flax NNX (Part 2)
62
Google Developers2.57 млн
Следующее
3 дня – 358:45Популярные
5 дней – 1752:02
5 дней – 1552:57Опубликовано 4 декабря 2025, 5:00
You know that debugging is crucial when doing any kind of software development. JAX and Flax NNX offer amazing performance and a clean API, but debugging can feel different due to JAX's compilation model. Today, we'll bridge the gap, exploring how to debug JAX and NNX effectively, relating techniques back to your PyTorch experience. We'll focus specifically on the Flax NNX API. In this second episode of our three part series on debugging we’ll build on the first by introducing more advanced and specialized techniques, including how to temporarily disable JIT and how to leverage the powerful, built-in inspection tools provided by Flax NNX.
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
Свежие видео
3 дня – 4 1881:23:36
3 дня – 1076:18
4 дня – 6 5290:15
4 дня – 74 7841:24
4 дня – 531:17Случайные видео
1 день – 7 5801:13
312 дней – 23 2422:32
03.05.24 – 86 9628:58
11.11.22 – 95 5472:20
13.01.22 – 11 5171:30
09.01.15 – 142 2574:26
1 день – 87 76925:45
3 дня – 1 5500:18
3 дня – 319:39
4 дня – 3761:41
5 дней – 5 9681:11
8 дней – 5 3490:53
7 часов – 14 69919:23