This repository was archived by the owner on Nov 7, 2024. It is now read-only.
This repository was archived by the owner on Nov 7, 2024. It is now read-only.
Add sparse linalg to Jax? #785
Description
Hi all,
Per the conversation in jax-ml/jax#4025 the Jax devs are interested in supporting at least some of the sparse linalg operations we've spent so much time on. Thus we may have more impact by migrating some of our eigs/eigsh/gmres code into Jax directly.