Introducing Flax Nnx (part 2)

2026 11:41
Synopsis
This episode is a deep dive into making NNX code fast, and understanding its relationship with the core JAX functional paradigm.

You Might Also Like

Introducing Flax NNX (Part 1) 10:32
Introducing Flax Nnx (part 1)
2026
Debugging JAX & Flax NNX (Part 2) 12:34
Debugging Jax & Flax Nnx (part 2)
2026
Optimizing Flax NNX Models with Optax (Part 2) 12:02
Optimizing Flax Nnx Models With Optax (part 2)
2026
Debugging JAX & Flax NNX (Part 1) 6:42
Debugging Jax & Flax Nnx (part 1)
2026
Why JAX and Flax NNX? 11:11
Why Jax And Flax Nnx?
2026
NNX the neural network for Flax explained 2:44
Nnx The Neural Network For Flax Explained
2026
JAX DevLab 2025 - What's new for Flax NNX? 8:29
Jax Devlab 2025 - What's New For Flax Nnx?
2026
Flax Milk 101: What You Need to Know 2:48
Flax Milk 101: What You Need To Know
2023
What Is The Difference Between Flaxseed vs. Linseed? 3:26
What Is The Difference Between Flaxseed Vs. Linseed?
2024
Enhancing Reliability (Part 2) 14:29
Enhancing Reliability (part 2)
2026
Introducing Tunix: JAX-Native LLM Post-Training Library. Flax NNX and Tunix. (SFT), (RL), TPU, PPO 14:04
Introducing Tunix: Jax-native Llm Post-training Library. Fla...
2026
Download Options
Choose a download method below. All links open in new tabs.
Service Features Action
SaveFrom
MP4 & MP3 • HD Quality • Browser Extension Available
Download
Security Notice: These are third-party services. We recommend using antivirus software and being cautious of pop-up ads.