Sponsored
View Detailed Profile
Debugging JAX & Flax NNX (Part 2)

Debugging JAX & Flax NNX (Part 2)

You know that

Debugging JAX & Flax NNX (Part 1)

Debugging JAX & Flax NNX (Part 1)

You know that

Debugging JAX & Flax NNX (Part 3)

Debugging JAX & Flax NNX (Part 3)

You know that

Introducing Flax NNX (Part 2)

Introducing Flax NNX (Part 2)

This episode is a deep dive into making

Optimizing Flax NNX Models with Optax (Part 2)

Optimizing Flax NNX Models with Optax (Part 2)

This is

Sponsored
Checkpointing Flax NNX Models with Orbax (Part 2)

Checkpointing Flax NNX Models with Orbax (Part 2)

In

Optimizing Flax NNX Models with Optax (Part 1)

Optimizing Flax NNX Models with Optax (Part 1)

This is the first of our two

Introducing Flax NNX (Part 1)

Introducing Flax NNX (Part 1)

We'll be diving into

Intro to JAX Debugging (Prints, Breakpoints, and Callbacks)

Intro to JAX Debugging (Prints, Breakpoints, and Callbacks)

Learn powerful

Checkpointing Flax NNX Models with Orbax (Part 1)

Checkpointing Flax NNX Models with Orbax (Part 1)

Orbax is the standard checkpointing library in the

Introducing Flax NNX (Part 3)

Introducing Flax NNX (Part 3)

In this episode we'll continue our look at

Understanding JAX: JIT, XLA, and Pure Functions Explained

Understanding JAX: JIT, XLA, and Pure Functions Explained

Are you exploring

JAX/OpenXLA DevLab 2025 - Performance Debugging with XProf

JAX/OpenXLA DevLab 2025 - Performance Debugging with XProf

JAX