torchdiffeq
torchdiffeq copied to clipboard
Feature Request: differentiable event handling for scipy solvers
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.