Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax - View it on GitHub