diffrax icon indicating copy to clipboard operation
diffrax copied to clipboard

Published 20 hours ago • verified publisher icon patrick-kidger

Have SDEs using `RecursiveCheckpointAdjoint` rather than `DirectAdjoint`

Open patrick-kidger opened this issue 1 year ago • 0 comments

This relies on JAX implementing a way to detect symbolic zero tangents to a custom_vjp.

patrick-kidger avatar Feb 20 '23 03:02 patrick-kidger