Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

Module: tfp.experimental.substrates.jax.optimizer

View source on GitHub

TensorFlow Probability Optimizer python package.

Modules

linesearch module: Line-search optimizers package.