Additon of levenberg-marquardt optimizer in TORCH.OPTIM #51407
Labels
enhancement
Not as big of a feature, but technically not a bug. Should be easy to fix
module: optimizer
Related to torch.optim
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
馃殌 Feature
Additon of levenberg-marquardt algorithm in TORCH.OPTIM
Motivation
The Levenberg-Marquardt (LM) algorithm is a classical and standard algorithm used to solve non-linear least squares problems. Though the scipy.optimize.curve_fit provides a great implementation of it, it is serial and runs of CPU. The curve_fit from SciPy becomes very slow in case of large problems, and therefore a GPU version is needed.
Pitch
GPU version of scipy.optimize.curve_fit
cc @vincentqb
The text was updated successfully, but these errors were encountered: