@vsv numba поддерживает не все функции из numpy, поэтому приходилось писать свой алгоритм на питоне и оборачивать его в numba. По производительности, кстати, получилось не сильно медленее, чем если бы использовали numpy.