Module: tfp.substrates.jax.optimizer.convergence_criteria

TensorFlow Probability convergence criteria for optimizations.

Classes

class ConvergenceCriterion: Base class for stopping rules.

class LossNotDecreasing: Simple convergence criterion based on lack of decrease in loss values.

class SuccessiveGradientsAreUncorrelated: Convergence criterion based on inner products between successive gradients.