large speedup using the numba JIT compiler