Published
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 → https://goo.gle/learning-jax
Subscribe to Google for Developers → https://goo.gle/developers
Speaker: Robert Crowe
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
Up Next
-
10:52
MSI GS30 Shadow - Part Beastly Gaming Rig, Part Ultrabook
-
03:29
Part hot hatch, part eco car: 2015 Volkswagen Golf GTE
-
13:12
Checkpointing Flax NNX Models with Orbax (Part 1)
-
12:02
Optimizing Flax NNX Models with Optax (Part 2)
-
08:12
Demo: JAX, Flax and Gemma
-
10:32
Introducing Flax NNX (Part 1)
-
11:25
GTC 2015: Introducing the GeForce GTX TITAN X GPU (part 3)
-
08:12
Demo: Gemma 2 architecture: JAX, Flax, and more
-
09:39
Debugging JAX & Flax NNX (Part 3)
-
06:03
Introducing Flax NNX (Part 3)
-
05:14
Watch This Skateboarding, Slacklining Robot ???? : Part Humanoid, Part Drone
-
01:33
RoboCop: Rogue City - Official Part Man, Part Machine Trailer | gamescom 2023
-
01:24
The Polaroid Snap is part digital camera, part printer
-
03:54
Mercedes Vision Urbanetic First Look: Part taxi, part delivery van
-
13:22
Introducing The Witcher 3: Wild Hunt (Part 1 of 3)
-
12:13
Checkpointing Flax NNX Models with Orbax (Part 2)
-
11:11
Why JAX and Flax NNX?
-
08:45
Optimizing Flax NNX Models with Optax (Part 1)
-
03:17
Black Mirror: Bandersnatch Is Part Video Game, Part Movie
-
17:36
GTC 2015: Teaching Deep Neural Networks, Introducing DIGITS DevBox (part 6)
-
06:42
Debugging JAX & Flax NNX (Part 1)
-
12:34
Debugging JAX & Flax NNX (Part 2)
Loading...