optax.contrib.DoGState# class optax.contrib.DoGState( is_init_step: Array, init_params: optax.ArrayTree, max_dist: Array, sum_sq_norm_grads: Array, )[source]# State for DoG optimizer.
