Module: tfp.substrates.jax.optimizer

TensorFlow Probability Optimizer python package.

Modules

linesearch module: Line-search optimizers package.

Functions

bfgs_minimize(...): Applies the BFGS algorithm to minimize a differentiable function.

converged_all(...): Condition to stop when all batch members have converged or failed.

converged_any(...): Condition to stop when any batch member converges, or all have failed.

lbfgs_minimize(...): Applies the L-BFGS algorithm to minimize a differentiable function.

nelder_mead_minimize(...): Minimum of the objective function using the Nelder Mead simplex algorithm.

nelder_mead_one_step(...): A single iteration of the Nelder Mead algorithm.