Sparse Matrices 5: I bind you Nancy
Read OriginalThis technical article details the process of adding new primitives to JAX for differentiable sparse linear algebra. It covers implementing abstract and concrete evaluation rules for functions like matrix solves, using a 2D Laplacian as a test case, and lays groundwork for future posts on JIT compilation and derivatives.
Comments
No comments yet
Be the first to share your thoughts!
Browser Extension
Get instant access to AllDevBlogs from your browser