Sparse matrices 6: To catch a derivative, first you’ve got to think like a derivative
Read OriginalThis 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.
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
The Beautiful Web
Jens Oliver Meiert
•
2 votes
2
When your coding agent doesn’t understand your project, you’ll get junk
Benjamin Cane
•
1 votes
3
LLM Use in the Python Source Code
Miguel Grinberg
•
1 votes
4
Wagon’s algorithm in Python
John D. Cook
•
1 votes
5
An example conversation with Claude Code
Dumm Zeuch
•
1 votes