def cond(args): i, lower, _, upper = args return jnp.logical_and(jnp.less(i, max_it), jnp.less(abs_tol, jnp.amax(upper - lower)))
def cond(iqc): i, q, count = iqc return jnp.less(i, n)
def cond(args): i, lower, _, upper = args return jnp.logical_and(jnp.less(i, max_it), jnp.less(abs_tol, jnp.amax(upper - lower)))
def cond(iqc): i, q, count = iqc return jnp.less(i, n)