jrenaud90 / numba_special

Numba overloads for SciPy's special functions

Numba Special

Numba special allows SciPy's special functions to be used in Numba.

How to use it

>>> import numba
>>> import scipy.special as sc
>>> import numba_special  # The import generates Numba overloads for special
>>> @numba.njit
... def gamma_plus_1(x):
...     return sc.gamma(x) + 1.0
...
>>> gamma_plus_1(5.0)
25.0

For detailed information on which functions can be used from Numba, check the documentation.

Installing

Numba special depends on SciPy and Numba. Until the fix for this issue is included in a Numba release, numba_special must be installed against the master branch of Numba.