Open in PodwiseOpen

Podcast cover

Google for Developers · Technology

Google for Developers

Subscribe to join a community of creative developers and learn the latest in Google technology — from AI and cloud, to mobile and web. Explore more at developers.google.com

Episodes

Optimizing Flax NNX Models with Optax (Part 2)

NumPy & JAX NumPy (Part 2)

Scaling Up (Part 1)

Getting started with MNIST

Scaling Up (Part 2)

Debugging JAX & Flax NNX (Part 1)

Enhancing Reliability (Part 1)

Efficient Data Loading

NumPy & JAX NumPy (Part 1)

Why JAX and Flax NNX?

Serving JAX Models with vLLM & SGLang

Introducing Flax NNX (Part 3)

Checkpointing Flax NNX Models with Orbax (Part 2)

Debugging JAX & Flax NNX (Part 3)

Introducing Flax NNX (Part 2)

Scaling Up (Part 3)

Introducing Flax NNX (Part 1)

JAX AI Stack: Summary & Conclusion

Optimizing Flax NNX Models with Optax (Part 1)

Checkpointing Flax NNX Models with Orbax (Part 1)

Page 7

Follow this podcast in Podwise

Sign in to get AI summaries, transcripts and mind maps for any episode, including new ones.

Open in Podwise