Skip to main content
Back to top
Ctrl
+
K
🚀 Getting started
🖼️ Example gallery
Examples
Adversarial training
ResNet on CIFAR10 with Flax and Optax.
Simple NN with Flax.
Gradient Accumulation
Lookahead Optimizer on MNIST
Meta-Learning
MLP MNIST
Character-level Transformer on Tiny Shakespeare
Optimistic Gradient Descent in a Bilinear Min-Max Problem
Contrib Examples
Differentially private convolutional neural network on MNIST.
Reduce on Plateau Learning Rate Scheduler
Sharpness-Aware Minimization (SAM)
🛠️ Development
📖 Reference
Optimizers
Transformations
Combining Optimizers
Optimizer Wrappers
Optimizer Schedules
Apply Updates
Projections
Losses
Control Variates
Stochastic Gradient Estimators
Utilities
🔧 Contrib
Search
Error
Please activate JavaScript to enable the search functionality.
Ctrl
+
K