def fun_ref(x, y):
    if (x < 3):
        return (2.0 * jnp.sin(y))
    else:
        return (2.0 * jnp.cos(x))
def fun_ref(x, y):
    if (x < 3):
        return (2.0 * jnp.sin(y))
    else:
        return (2.0 * jnp.cos(x))