jax.scipy.special.polygamma#

jax.scipy.special.polygamma(n, x)[source]#

The polygamma function.

JAX implementation of scipy.special.polygamma().

\[\mathrm{polygamma}(n, x) = \psi^{(n)}(x) = \frac{\mathrm{d}^{n+1}}{\mathrm{d}x^{n+1}} \log \Gamma(x)\]

where \(\psi\) is the digamma() function and \(\Gamma\) is the gamma() function.

Parameters:

  • n (ArrayLike) – arraylike, integer-valued. The order of the derivative.

  • x (ArrayLike) – arraylike, real-valued. The value at which to evaluate the function.

Returns:

array

Return type:

Array

See also