optax.contrib.MadgradState# class optax.contrib.MadgradState(count: Array, grad_sum_sq: optax.Updates, s: optax.Updates, x0: optax.Params)[source]# State for the MADGRAD optimizer.
