Ë ñukiêã óŒ—ddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z mZdded ed ed e fd „Zdded ed ed e fd „Zy)éN)Úlax)Únumpy)Ú_const)Úpromote_args_inexact)Úxlog1py)ÚArrayÚ ArrayLikeÚkÚpÚlocÚreturncód—td|||«\}}}t|d«}t|d«}tj||«}t tj||«| «tj |«z}t jtj||«tj |«S)aòGeometric log probability mass function. JAX implementation of :obj:`scipy.stats.geom` ``logpmf``. The Geometric probability mass function is given by .. math:: f(k) = (1 - p)^{k-1}p for :math:`k\ge 1` and :math:`0 \le p \le 1`. Args: k: arraylike, value at which to evaluate the PMF p: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter Returns: array of logpmf values. See Also: :func:`jax.scipy.stats.geom.pmf` z geom.logpmfré) rÚ _lax_constrÚsubrÚlogÚjnpÚwhereÚleÚnpÚinf)r r r ÚzeroÚoneÚxÚ log_probss úT/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/scipy/stats/geom.pyÚlogpmfrs‰€ô0# =°!°Q¸Ó<)€!€QˆÜ AqÓ €$Ü1aÓ€#Ü ‡ggˆaƒo€!Ü”c—g‘g˜a “o¨ rÓ*¬S¯W©W°Q«ZÑ7€)Ü ‰”3—6‘6˜!˜T“?¤R§V¡V G¨YÓ 7Ð7ócóB—tjt|||««S)aëGeometric probability mass function. JAX implementation of :obj:`scipy.stats.geom` ``pmf``. The Geometric probability mass function is given by .. math:: f(k) = (1 - p)^{k-1}p for :math:`k\ge 1` and :math:`0 \le p \le 1`. Args: k: arraylike, value at which to evaluate the PMF p: arraylike, distribution shape parameter loc: arraylike, distribution offset parameter Returns: array of pmf values. See Also: :func:`jax.scipy.stats.geom.logpmf` )rÚexpr)r r r s rÚpmfr!9s€ô0 ‰”˜˜1˜cÓ"Ó #Ð#r)r)rrÚjax._srcrrÚjax._src.lax.laxrrÚjax._src.numpy.utilrÚjax._src.scipy.specialrÚjax._src.typingrr rr!©rrúr(s_ðóåÝ!Ý1Ý4Ý*ß,ñ8ˆið8˜Ið8¨Ið8¸eó8ñ@$ˆ9ð$˜ð$¨ð$¸5ô$r