Add some variants using the Numba package to JIT the Python loop.