My spontaneous recommendation would be to model everything with a JuMP.jl program (either linear or convex) and then use DiffOpt.jl to differentiate it with respect to the parameters. But maybe @mbesancon would suggest something else.
You can draw inspiration from GraphsOptim.jl on how to model the flow in JuMP.
1 Like