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
Copy file name to clipboardExpand all lines: docs/further_details/faq.md
+15Lines changed: 15 additions & 0 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -8,6 +8,21 @@ If you're using a Runge--Kutta method like [`diffrax.Dopri5`][] etc., then try s
8
8
9
9
Try switching to 64-bit precision. (Instead of the 32-bit that is the default in JAX.) [See here](https://jax.readthedocs.io/en/latest/notebooks/Common_Gotchas_in_JAX.html#double-64bit-precision).
10
10
11
+
### How does this compare to `jax.experimental.ode.odeint`?
In practice, `TSit5` is usually a better solver than `Dopri5`. And the default adjoint method (`RecursiveCheckpointAdjoint`) is usually a better choice than `BacksolveAdjoint`.
25
+
11
26
### I'm getting a `CustomVJPException`.
12
27
13
28
This can happen if you use [`diffrax.BacksolveAdjoint`][] incorrectly.
0 commit comments