flax.optim package#

Optimizer Base Classes#

MultiOptimizer#

Available optimizers#