5/29/2022
•
EN
Sparse matrices 6: To catch a derivative, first you’ve got to think like a derivative
Part 6 of a series on making sparse linear algebra differentiable in JAX, focusing on implementing Jacobian-vector products for custom primitives.