Getting started with MNIST

From the creator

Now that you’ve learned the basic concepts of the JAX AI Stack, let’s work through everybody’s favorite ML example - MNIST - to deepen your understanding. We will of course be training a simple CNN with NNX. Resources: Learn more → https://goo.gle/learning-jax Subscribe to Google for Developers → https://goo.gle/developers Speaker: Robert Crowe

Published 2025-12-04 · Watch on YouTube ↗
Availability and playback restrictions are controlled by the original channel.