Use just the input types in the jit compiling step