ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

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(...)