Gradient of loss function from ODE solution

[Enrico_Camporeale]

@ChrisRackauckas Could you please give a look at this whenever you got the chance? It should be really straightforward and I have tried to follow your example here (just a slightly more complicated ODE)

Correcting ODE initial conditions using automatic differentiation

Hi all, I’m trying to find a way to correct initial conditions of a system of ODEs by using automatic differentiation through the ODE solver. I have implemented a similar algorithm in Python using TorchDIffEq but I’m new to Julia and having trouble debugging the errors. I’ve gone through the docs and examples of both Zygote and ReverseDiff (which I think are best suited to my purpose) but I’m having difficulty adapting them to my problem. What I want to do is first solve the system of ODEs with…

Something is obviously escaping me
Many thanks!