gwkokab.inference.numpyro_discrete_poisson_likelihood
=====================================================

.. py:module:: gwkokab.inference.numpyro_discrete_poisson_likelihood


Functions
---------

.. autoapisummary::

   gwkokab.inference.numpyro_discrete_poisson_likelihood.numpyro_discrete_poisson_likelihood


Module Contents
---------------

.. py:function:: numpyro_discrete_poisson_likelihood(dist_fn: collections.abc.Callable[Ellipsis, numpyro.distributions.Distribution], priors: gwkokab.models.utils.JointDistribution, variables: Dict[str, numpyro.distributions.Distribution], variables_index: Dict[str, int], poisson_mean_estimator: collections.abc.Callable[[gwkokab.models.utils.ScaledMixture], jax.Array], where_fns: Optional[List[collections.abc.Callable[Ellipsis, jax.Array]]], constants: Dict[str, jax.Array], variance_cut_threshold: float | None) -> collections.abc.Callable[[Tuple[jax.Array, Ellipsis], Tuple[jax.Array, Ellipsis], Tuple[jax.Array, Ellipsis]], jax.Array]