Когда код Numba разочаровывает, виноват почти никогда не компилятор, а обычно граница вокруг скомпилированного кода: либо она не пересекается, либо недостаточно широка, либо пересекается при каждом запуске.