def cond_tf_func(*args: TfVal) -> TfVal:
    (pred,) = _interpret_jaxpr(cond_jaxpr, *cond_consts, *args)
    return pred
def cond_tf_func(*args: TfVal) -> TfVal:
    (pred,) = _interpret_jaxpr(cond_jaxpr, *cond_consts, *args)
    return pred