quantax.optimizer.PlainUpdater

class quantax.optimizer.PlainUpdater

Bases: Updater

The plain update \(\dot\theta = \mathrm{solve}(\bar O, \bar\epsilon)\).

__init__()

init(nparams: int) → dict[str, Array]

Initialize the persistent buffers for nparams parameters.

update(core_solve: Callable[[...], Array], Obar: Array, Ebar: Array, buffers: dict[str, Array]) → tuple[Array, dict[str, Array]]

Generate the optimization step and the new buffers, given the equation solver core_solve and the current buffers.