You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Diffrax is a JAX ODE solving library providing access to many different solvers, and more control over how computations (like back propagation) are done. It also allegedly offers speed improvements over JAX's odeint function for the same method and tolerances.
What is the expected behavior?
Add access to solvers from this library through solve_ode.
The text was updated successfully, but these errors were encountered:
Diffrax is a JAX ODE solving library providing access to many different solvers, and more control over how computations (like back propagation) are done. It also allegedly offers speed improvements over JAX's
odeint
function for the same method and tolerances.What is the expected behavior?
Add access to solvers from this library through
solve_ode
.The text was updated successfully, but these errors were encountered: