Vpf+ ddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZmZddeded ed ed ef d Zddeded ed ed ef d ZdS))laxN)_const)promote_args_inexact)gammalnxlogy)Array ArrayLikeknplocreturnc td||||\}}}}t|d}tj||}tjtjt tj||t |t tj||}tjt ||t |tj||}tj||}tjtj ||tj |S)aINegative-binomial log probability mass function. JAX implementation of :obj:`scipy.stats.nbinom` ``logpmf``. The negative-binomial probability mass function is given by .. math:: f(k) = {{k+n-1} \choose {n-1}}p^n(1-p)^k for :math:`k \ge 0` and :math:`0 \le p \le 1`. Args: k: arraylike, value at which to evaluate the PMF n: arraylike, distribution shape parameter p: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter Returns: array of logpdf values. See Also: :func:`jax.scipy.stats.nbinom.pmf` z nbinom.logpmf) r _lax_constrsubraddrjnpwhereltinf) r r r r oney comb_termlog_linear_term log_probss [/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/scipy/stats/nbinom.pylogpmfrs2&oq!QDD,!Q31a# gaoo!gGGCGAqMM " "GAJJ//C1I1I)GE!QKKq#'#q//)B)BCC/gi11) 36!S>>CG8Y 7 77cJtjt||||S)aBNegative-binomial probability mass function. JAX implementation of :obj:`scipy.stats.nbinom` ``pmf``. The negative-binomial probability mass function is given by .. math:: f(k) = {{k+n-1} \choose {n-1}}p^n(1-p)^k for :math:`k \ge 0` and :math:`0 \le p \le 1`. Args: k: arraylike, value at which to evaluate the PMF n: arraylike, distribution shape parameter p: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter Returns: array of pmf values. See Also: :func:`jax.scipy.stats.nbinom.logpmf` )rexpr)r r r r s rpmfr";s"2 1a%% & &&r)r)jaxr jax.numpynumpyrjax._src.lax.laxrrjax._src.numpy.utilrjax._src.scipy.specialrrjax._src.typingrr rr"rrr+s11111144444411111111,,,,,,,,!8!8i!8I!8)!8)!8E!8!8!8!8H''9''y'y'''''''r