Google I/O returns May 18-20! Reserve space and build your schedule Register now

Module: tfp.substrates.jax.optimizer

TensorFlow Probability Optimizer python package.


linesearch module: Line-search optimizers package.


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.