optax.update_infinity_moment

optax.update_infinity_moment#

optax.update_infinity_moment(
updates,
moments,
decay,
eps,
)[source]#

Compute the exponential moving average of the infinity norm.