The jit._fork seems promising, thanks! It wasn’t clear to me from the documentation, but does the fork and wait have to be called within a JitScriptModule or can it be called on functions from JitScriptModules in regular Python code?