optax.AddNoiseState

Contents

optax.AddNoiseState#

class optax.AddNoiseState(count: Array, rng_key: Array)[source]#

State for adding gradient noise. Contains a count for annealing.