search

patrick-kidger / lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
Apache License 2.0
365 stars 24 forks source link

Add support for scalar linear solves (via division) to AutoLinearSolver. #30

Open patrick-kidger opened 1 year ago