AbstractUnconstrainedMinimizer

class pmrf.optimize.AbstractUnconstrainedMinimizer

Bases: Module

Abstract interface for unconstrained minimization algorithms.

abstractmethod run(fn: Callable[[PyTree, Any], Shaped[jaxlib._jax.Array, '']], y0: PyTree, args: Any, max_iter: int = 1024, **kwargs) → tuple[MinimizeResult, PyTree]

Execute the minimization algorithm.

Parameters:

• fn (callable) – The objective function to minimize.

• y0 (PyTree) – The initial parameter guess.

• args (Any) – Args to pass to fn.

• max_iter (int = 1024) – The maximum number of iterations to take.

• **kwargs – Runtime arguments forward to the solver backend.

Returns:

A tuple of (pmrf.optimize.MinimizeResult, metrics)`.

Return type:

tuple