jax.scipy.stats.nbinom.pmf#

jax.scipy.stats.nbinom.pmf(k, n, p, loc=0)[source]#

Negative-binomial probability mass function.

JAX implementation of scipy.stats.nbinom pmf.

The negative-binomial probability mass function is given by

\[f(k) = {{k+n-1} \choose {n-1}}p^n(1-p)^k\]

for \(k \ge 0\) and \(0 \le p \le 1\).

Parameters:

k (Array | ndarray | bool_ | number | bool | int | float | complex) – arraylike, value at which to evaluate the PMF
n (Array | ndarray | bool_ | number | bool | int | float | complex) – arraylike, distribution shape parameter
p (Array | ndarray | bool_ | number | bool | int | float | complex) – arraylike, distribution shape parameter
loc (Array | ndarray | bool_ | number | bool | int | float | complex) – arraylike, distribution offset parameter

Returns:

array of pmf values.

Return type:

Array

jax.scipy.stats.nbinom.pmf