Contrib Examples# Examples that make use of the optax.contrib module. Differentially private convolutional neural network on MNIST. Using the Muon Optimizer in Optax Reduce on Plateau Learning Rate Scheduler Recreate AdeMAMix Rosenbrock Plot from Paper Sharpness-Aware Minimization (SAM)
