jax-linear-operator (pypi): A JAX linear operator library.
Keywords: jax|linear-operator|low-rank|linear-algebra