ops

jaxns.framework.ops

Module Contents

simulate_prior_model(key, prior_model)

Simulate a prior model.

Parameters:

• key (jaxns.internals.types.PRNGKey) – PRNGKey

• prior_model (jaxns.framework.bases.PriorModelType) – A prior model

Returns:

a tuple of the likelihood input variables, and dict of non-hidden (named) prior variables.

Return type:

Tuple[jaxns.internals.types.LikelihoodInputType, jaxns.internals.types.XType]