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
Profiling Pytorch/XLA on TPUs with XProf
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)
Debugging JAX & Flax NNX (Part 2)
Enhancing Reliability (Part 2)
Leveraging the JAX AI Stack
Stunning Visualizations with AI-First Colab
How to secure your AI Agents: A Technical Deep-dive
GDG Summit MENA-T
How do thinking and reasoning models work?
Simplify framework updates with Google Antigravity
End to End Machine Learning with AI First Colab