RSVP for your your local TensorFlow Everywhere event today!

tfp.substrates.jax.optimizer.converged_all

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