gwkokab.inference.flowMC_analytical_poisson_likelihood

Functions

flowMC_analytical_poisson_likelihood(...)

Module Contents

gwkokab.inference.flowMC_analytical_poisson_likelihood.flowMC_analytical_poisson_likelihood(dist_fn: Callable[Ellipsis, numpyro.distributions.Distribution], priors: gwkokab.models.utils.JointDistribution, variables: Dict[str, numpyro.distributions.Distribution], constant_params: Dict[str, Any], variables_index: Dict[str, int], poisson_mean_estimator: Callable[[gwkokab.models.utils.ScaledMixture], tuple[jaxtyping.Array, jaxtyping.Array]], variance_cut_threshold: float | None) → Callable[[jaxtyping.Array, Dict[str, Any]], jaxtyping.Array]