You can add @jit.ignore decorated forwarders.
Alternatively, C++ extension ops work with JIT, even supporting gradients, but I don’t know how exactly to register them from python. In C++ it is like:
static auto registry =
torch::RegisterOperators()
.op("namespace::func", &func)

with that you can call torch.ops.namespace.func(…)