Skip to main content
Back to top
Ctrl
+
K
Search
Ctrl
+
K
🚀 Getting started
🖼️ Example gallery
Examples
Adversarial training
ResNet on CIFAR10 with Flax NNX and Optax.
Simple NN with Flax.
Freezing Parameters in Optax
Gradient Accumulation
Summary
L-BFGS
Linear assignment problem
Lookahead Optimizer on MNIST
Meta-Learning
MLP MNIST
Character-level Transformer on Tiny Shakespeare
Optimistic Gradient Descent in a Bilinear Min-Max Problem
Perturbed optimizers
Contrib Examples
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)
🛠️ Development
📖 Reference
Assignment problem
Optimizers
Transformations
Combining Optimizers
Optimizer Wrappers
Optimizer Schedules
Apply Updates
Perturbations
Projections
Losses
Utilities
Microbatching
🔧 Contrib
🧪 Experimental
.rst
.pdf
🧪 Experimental
🧪 Experimental
#
Experimental features subject to changes before being graduated into
optax
.
