Allow vectorize to work on functions with jit dectorator