jax.scipy.stats.binom.pmf#

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

Binomial probability mass function.

JAX implementation of scipy.stats.binom pmf.

The binomial probability mass function is defined as

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

for \(0\le p\le 1\) and non-negative integers \(k\).

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.binom.pmf