optax.contrib.DifferentiallyPrivateAggregateState

class optax.contrib.DifferentiallyPrivateAggregateState(

rng_key: Array,

)

State containing PRNGKey for differentially_private_aggregate.