tfp.substrates.jax.optimizer.converged_any

Condition to stop when any batch member converges, or all have failed.