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.
