Adds an option to support forward-mode automatic differentiation in all minimisers#114

Merged

patrick-kidger merged 7 commits intopatrick-kidger:mainfrom johannahaffner:forward-fix

Feb 8, 2025

+120-34

Commits on Feb 2, 2025

Adding an option to enable forward-mode automatic differentiation for all minimisers.
Johanna Haffner
committed

Commits on Feb 3, 2025

improve docstring
Johanna Haffner
committed
move creation of unit pytrees to enable vmapping
Johanna Haffner
committed

Commits on Feb 6, 2025

remove custom _jacfwd, replace with eqx.filter_jacfwd
Johanna Haffner
committed
eqx.filter_jacfwd -> jax.jacfwd
Johanna Haffner
committed

Commits on Feb 7, 2025

mode -> autodiff_mode, adds forward-only test case
Johanna Haffner
committed
add a pyright: ignore flag for return type (sol.ys) of dfx.diffeqsolve
Johanna Haffner
committed