torchdiffeq icon indicating copy to clipboard operation
torchdiffeq copied to clipboard

Feature Request: differentiable event handling for scipy solvers

Open rtqichen opened this issue 4 years ago • 0 comments

Modify the wrapper (https://github.com/rtqichen/torchdiffeq/blob/master/torchdiffeq/_impl/scipy_wrapper.py) to take in an event_fn and wrap around the event functionality in scipy.solve_ivp.

rtqichen avatar Mar 23 '21 16:03 rtqichen