Module compute::optimize::optimizers[][src]

Expand description

Various optimization algorithms (eg. Adam, SGD, Levenberg-Marquardt).

Structs

Adam

Implements the Adam optimizer. See Kingma and Ba 2014 for details about the algorithm.

LM

Implements a Levenberg-Marquardt optimizer for solving (non-linear) least squares problems.

SGD

Implements the Stochastic Gradient Descent optimizer with (Nesterov) momentum.

Enums

GradFn

Traits

Optimizer