Skip to content

Commit

Permalink
Fix Hessian example
Browse files Browse the repository at this point in the history
  • Loading branch information
patrick-kidger committed Nov 9, 2023
1 parent 017758d commit 6192f62
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/hessian.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -78,9 +78,9 @@
"id": "a3ec6532-5b0a-4e4c-af33-bef58c0a7319",
"metadata": {},
"source": [
"Note the use of the `scan_kind` argument to `Tsit5`. By default, Diffrax internally uses constructs that are optimised specifically for first-order reverse mode autodifferentiation. This argument is needed to switch to a different implementation that is compatible with higher-order autodiff. (In this case: for the loop-over-stages in the Runge--Kutta solver.)\n",
"Note the use of the `scan_kind` argument to `Tsit5`. By default, Diffrax internally uses constructs that are optimised specifically for first-order reverse-mode autodifferentiation. This argument is needed to switch to a different implementation that is compatible with higher-order autodiff. (In this case: for the loop-over-stages in the Runge--Kutta solver.)\n",
"\n",
"In similar fashion, if using `saveat=SaveAt(steps=True)` then you will need to pass `adjoint=DirectAdjoint()`. (In this case: for the loop-over-saving output.)"
"In similar fashion, if using `saveat=SaveAt(ts=...)` (or a handful of other esoteric cases) then you will need to pass `adjoint=DirectAdjoint()`. (In this case: for the loop-over-saving output.)"
]
}
],
Expand Down

0 comments on commit 6192f62

Please sign in to comment.