JAX Mastery

Master JAX: XLA-compiled NumPy with autodiff. Learn jit, grad, vmap, pmap, sharding, and the patterns that power Google's most ambitious AI research.

6
Lessons
💻
Code Examples
Production-Ready
100%
Free

Lessons in This Topic

Work through these 6 lessons in order, or jump to whichever topic you need most.