Save the date! Google I/O returns May 18-20 Register now

Module: oryx.experimental.optimizers

Module for optimizers in Oryx.

Modules

optix module: Reimplementation of a subset of the optax library using Oryx's state system.

Functions

adam(...)

gradient_descent(...)

noisy_sgd(...)

optimize(...): Runs several iterations of optimization and returns the result.

rmsprop(...)

sgd(...)