changing autojit to jit in numba version, it's still slow...