quantax.optimizer.March#

class quantax.optimizer.March#

Bases: Updater

The MARCH update, a variant of SR with first and second order momentum stored in the phi and v buffers.

__init__(mu: float = 0.95, beta: float = 0.995, norm_clip: float | None = None)#
Parameters:

  • mu – The first order momentum factor.

  • beta – The second order momentum factor.

  • norm_clip – The maximum norm of the step to be accumulated in the first and second order momentum. If not None, the raw step will be clipped to this value.

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

Initialize the persistent buffers for nparams parameters.

update(core_solve, Obar, Ebar, buffers)#

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