Depends on why you want to have such a forwarder compiled. But basically you’ll have to either avoid python’s star syntax (there are multiple jit-compatible ways to group arguments, but they’re intrusive wrt base_module - i.e. all wrapped modules need to implement a special interface) or have some non-compiled code in your model.