Add hessian diagonal search support to LM optimizer (#45)

SimonBoothroyd · Nov 2, 2023 · 53bd12e · 53bd12e
1 parent 38f4133
commit 53bd12e
Show file tree

Hide file tree

Showing 3 changed files with 297 additions and 119 deletions.
diff --git a/descent/optim/__init__.py b/descent/optim/__init__.py
@@ -1,5 +1,5 @@
 """Custom parameter optimizers."""
 
-from descent.optim._lm import LevenbergMarquardt, LevenbergMarquardtConfig
+from descent.optim._lm import LevenbergMarquardtConfig, levenberg_marquardt
 
-__all__ = ["LevenbergMarquardt", "LevenbergMarquardtConfig"]
+__all__ = ["LevenbergMarquardtConfig", "levenberg_marquardt"]