Understanding Autodiff with JAX by Srihari Radhakrishna

radx.insite

Understand how autodiff works using JAX.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   Jax in Action manning.com

   site category #jax link ↗

   A hands-on guide to using JAX for deep learning and other mathematically-intensive applications.

2. ▲
   Plugging Into JAX by Nick Doiron medium.com

   site category #jax link ↗

   Compares Flax, Haiku, and Objax on the Kaggle flower classification challenge.

3. ▲
   DiffeRT github.com

   tool category #jax link ↗

   Differentiable Ray Tracing toolbox for Radio Propagation powered by the JAX ecosystem.

4. ▲
   JAX GitHub Discussions github.com

   site category #jax link ↗
5. ▲
   autodiff github.com

   tool category #cpp link ↗

   A modern, fast and expressive C++ library for automatic differentiation. [MIT] website

6. ▲
   Diffrax github.com

   tool category #jax link ↗

   Numerical differential equation solvers in JAX.

JAX › Tutorials and Blog Posts: “Understand how autodiff works using JAX.”