4 ms·Autodiff Cookbook: Part 1 – easy and powerful automatic differentiation in JAX2 points by sebg 6y ago