patrick-kidger / lineax

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

Fixed a crash with symbolic zero cotangents; fixed a crash with pytrees+CG #49

Closed patrick-kidger closed 1 year ago