Interpret scalar transformations as diagonal

patrick-kidger / lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

Apache License 2.0

365 stars 24 forks source link

Closed NeilGirdhar closed 3 months ago

NeilGirdhar commented 3 months ago

Would you be able to add some tests for this edge case?

Done

patrick-kidger commented 3 months ago

LGTM - merged! :)