Dan Simpson 5/19/2022

Sparse Matrices 5: I bind you Nancy

Read Original

This 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.

Sparse Matrices 5: I bind you Nancy

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