Save the date! Google I/O returns May 18-20 Register now

tfp.substrates.jax.optimizer.converged_all

Condition to stop when all batch members have converged or failed.