autodiff 2 What JAX Traces, and What It Refuses Jul 28, 2026 Three Ways to Take a Gradient: Tape, Trace, and Source Transform Jun 24, 2026