optax.AddNoiseState# class optax.AddNoiseState(count: Array, rng_key: Array)[source]# State for adding gradient noise. Contains a count for annealing.