I know this thread is a bit old, but for anyone still looking for a Levenberg-Marquardt implementation in PyTorch, I’ve developed one: torch-levenberg-marquardt. Hope it helps anyone who comes across this!