another nice speedup by calling the numba function directly