Introducing Flax NNX (Part 2)
2 223
17.6
Next
39 days – 1219:39Popular
26 days – 8 3925:00
138 days – 1 7152:22Published on 4 Dec 2025, 5:00
This episode is a deep dive into making NNX code fast, and understanding its relationship with the core JAX functional paradigm. It explains jit, pure functions, and the crucial role of NNX's transformation wrappers.
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
Fresh videos
3 days – 30 9701:27
5 days – 64 52811:43
6 days – 3 0920:41
15 days – 1 0734:29
20 days – 1390:15
21 day – 314 4395:22Random videos
08.12.24 – 1830:36
11.05.24 – 1 677 3716:26
22.02.24 – 1 2162:39
25.10.23 – 215 0341:33
14.02.07 – 223 3751:03
3 days – 3051:06:50
4 days – 2 3150:46
4 days – 123 83414:25
5 days – 15 7701:09
6 days – 17 0591:42:55
