fix: use parallelized numba functions if possible

fix: use parallelized numba functions if possible #617