Debugging JAX & Flax NNX (Part 1)
46
Google Developers2.57 млн
Следующее
3 дня – 4711:10Популярные
23 дня – 31 8790:12
4 дня – 6 44013:26Опубликовано 4 декабря 2025, 5:01
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. This is the first episode of a three episode series on debugging. This episode introduces the core problem of JAX's JIT compilation and presents the most direct, fundamental tools for inspecting runtime values, introducing the basics for anyone new to JAX debugging.
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
Свежие видео
6 дней – 1000:26
7 дней – 1 1255:08
10 дней – 1070:23
11 дней – 1325:08
11 дней – 1356:49Случайные видео
269 дней – 00:13
26.05.24 – 150 99110:01
09.07.23 – 152 05812:39
19.02.23 – 374 73716:19
28.06.13 – 473 71927:01
3 дня – 54 68218:31
3 дня – 19 4001:49
3 дня – 319:39
4 дня – 1 4520:54
8 дней – 34 8669:15
8 дней – 921:00
11 дней – 1 920 93113:10