Books
In-depth explorations of topics I care about. Each book is a comprehensive, hands-on guide written to build real understanding — not just familiarity.
Scaling Deep Learning with JAX
In ProgressAn interactive and visual adaptation of the original JAX scaling book. Uses JAX.js for in-browser learning — run and experiment with JAX code directly in your browser. Covers fundamentals, automatic differentiation, parallelism primitives, and scaling models across TPU/GPU clusters.
Start reading