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 test for grad-of-vmap #102

Closed patrick-kidger closed 2 months ago

patrick-kidger commented 3 months ago

Added tests for #101.

Requires https://github.com/patrick-kidger/equinox/pull/795 (so these tests will fail right now)