Contrib Examples# Examples that make use of the optax.contrib module. Differentially private convolutional neural network on MNIST. Reduce on Plateau Learning Rate Scheduler Sharpness-Aware Minimization (SAM)