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