Differentiating through optimization problem

[avikpal]

See section 2.3 https://arxiv.org/pdf/2403.16341#page=7.12. Once you rewrite the problem, you can also use LeastSquaresOptim using LeastSquaresOptim.jl · NonlinearSolve.jl