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

Added some pytree handling helpers to lx.internal #42

Closed patrick-kidger closed 1 year ago