optax.contrib.DifferentiallyPrivateAggregateState

optax.contrib.DifferentiallyPrivateAggregateState#

class optax.contrib.DifferentiallyPrivateAggregateState(rng_key: Array)[source]#

State containing PRNGKey for differentially_private_aggregate.