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.
