torchdyn
torchdyn copied to clipboard
Neural Hybrid Automata tutorial
Tutorial for this paper. odeint_hybrid is already in.
@frankschae On our end, it would be good to extend the NHA tutorial with adjoint sensitivity through the events. ping since you're working on sensitivity + NHDE and you might be able to provide guidance on implementation tricks later on.
awesome. I still have some issues with respect to the additional contributions for implicit event functions but I'll let you know once I figure it out. :)
I think we have now all the terms, tried to summarize them/implementation steps in the post:
https://frankschae.github.io/post/bouncing_ball/
The PR for BacksolveAdjoint is:
https://github.com/SciML/DiffEqSensitivity.jl/pull/445