JITFunction now takes Lambda instead of ast