jax

efc1104c - have loops module generate same-argument jaxprs for single-operand cond