jax

34d75c2e - JVP rule for single-operand cond primitive