For this you’ll want to directly use SciMLSensitivity.jl and write the loss funciton.
https://sensitivity.sciml.ai/dev/