Dan Simpson • 5/29/2022

Sparse matrices 6: To catch a derivative, first you’ve got to think like a derivative

Read Original

This technical article details implementing differentiation rules (Jacobian-vector products) for custom JAX primitives related to sparse matrix operations. It is part of an ongoing series aimed at enabling differentiable sparse linear algebra for advanced statistical computing. The post dives into deriving and coding the derivative for a linear solve operation in a JAX-traceable manner.

0 comments
#linear algebra #sparse matrices #Jax
Sparse matrices 6: To catch a derivative, first you’ve got to think like a derivative

Comments

No comments yet

Be the first to share your thoughts!

Browser Extension

Get instant access to AllDevBlogs from your browser

Top of the Week

1
Quoting Thariq Shihipar
Simon Willison • 2 votes
2
Using Browser Apis In React Practical Guide
Jivbcoop • 2 votes
3
Top picks — 2026 January
Paweł Grzybek • 1 votes
4
In Praise of –dry-run
Henrik Warne • 1 votes
5
Deep Learning is Powerful Because It Makes Hard Things Easy - Reflections 10 Years On
Ferenc Huszár • 1 votes
6
Vibe coding your first iOS app
William Denniss • 1 votes
7
AGI, ASI, A*I – Do we have all we need to get there?
John D. Cook • 1 votes
8
Dew Drop – January 15, 2026 (#4583)
Alvin Ashcraft • 1 votes