lineax
lineax copied to clipboard
Added test for grad-of-vmap
Added tests for #101.
Requires https://github.com/patrick-kidger/equinox/pull/795 (so these tests will fail right now)