Refer to new numba function wrapper