optax.update_moment_per_elem_norm# optax.update_moment_per_elem_norm( updates, moments, decay, order, )[source]# Compute the EMA of the order-th moment of the element-wise norm.
