Examples#
This directory contains examples using the optax library.
- 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
- Hyperparameters and dataset download
- Data preparation
- NanoLM Model Definition
- State, Optimizer, and Loss Definition
- Model training
- Text generation
- Optimistic Gradient Descent in a Bilinear Min-Max Problem