gwkokab.inference.numpyro_discrete_poisson_likelihood

Functions

numpyro_discrete_poisson_likelihood(...)

Module Contents

gwkokab.inference.numpyro_discrete_poisson_likelihood.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: List[collections.abc.Callable[Ellipsis, jax.Array]] | None, 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]