search

patrick-kidger / diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Apache License 2.0
1.38k stars 124 forks source link

Adjoints: specify which are forward/reverse mode autodiff compatible #145

Closed patrick-kidger closed 1 year ago

patrick-kidger commented 2 years ago

For each adjoint method, it would be convenient to include a table specifying which are compatible with each kind of autodifferentiation.

patrick-kidger commented 1 year ago

Closing as this will be accomplished in the next version.