Why Jax And Flax Nnx

Why JAX and Flax NNX?

We'll explore the strengths of

Debugging JAX & Flax NNX (Part 1)

You know that debugging is crucial when doing any kind of software development.

Introducing Flax NNX (Part 1)

We'll be diving into

JAX DevLab 2025 - What's new for Flax NNX?

This lighting talk discusses what's next for

NNX the neural network for Flax explained

Tutorial (

Optimizing Flax NNX Models with Optax (Part 1)

This is the first of our two part series on optimizing models with Optax. Let's explore how to optimize

Debugging JAX & Flax NNX (Part 2)

You know that debugging is crucial when doing any kind of software development.

Introducing Flax NNX (Part 3)

In this episode we'll continue our look at

Introducing Flax NNX (Part 2)

This episode is a deep dive into making

Debugging JAX & Flax NNX (Part 3)

You know that debugging is crucial when doing any kind of software development.

JAX in 100 Seconds

Try Brilliant free for 30 days https://brilliant.org/fireship You'll also get 20% off an annual premium subscription

PyTorch vs TensorFlow vs JAX: The Ultimate Comparison

PyTorch vs TensorFlow vs

Optimizing Flax NNX Models with Optax (Part 2)

This is part two of our two episode series on optimizing models with Optax! This episode builds on the foundation of...

Get started with JAX using NNX 🦾🧠

NNX

JAX Tutorial: The Lightning-Fast ML Library For Python

In this video today, we take a look at

Checkpointing Flax NNX Models with Orbax (Part 1)

Orbax is the standard checkpointing library in the

Who uses JAX?

So, you know what

Why FLAX Could Be Your New Favorite Deep Learning Library for NN

Intro to