jax_privacy.optimizers.as_augmented_optimizer

jax_privacy.optimizers.as_augmented_optimizer(optimizer)[source]

Wraps a plain Optax optimizer with an identity pre-clipping transform.