Fix autojit wrapper to special-case numba types as autojit function arguments