quantax.optimizer.Spring#

class quantax.optimizer.Spring#

Bases: Updater

The SPRING update, a variant of SR with momentum stored in the phi buffer.

__init__(mu: float = 0.9, norm_clip: float | None = None)#
Parameters:

  • mu – The momentum factor.

  • norm_clip – The maximum norm of the step to be accumulated in 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.