initialisation

jaxns.nested_samplers.common.initialisation

Module Contents

create_init_state(key, num_live_points, max_samples, model, mesh=None)

Return an initial sample collection, that will be incremented by the sampler.

Parameters:

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

• num_live_points (int) – the number of live points

• max_samples (int) – the maximum number of samples

• model (jaxns.framework.bases.BaseAbstractModel) – the model to use.

• mesh (Optional[jax._src.mesh.Mesh])

Returns:

the initial live point collection state: the initial state

Return type:

live_point_collection

create_init_termination_register()

Initialise the termination register.

Returns:

The initial termination register.

Return type:

jaxns.nested_samplers.common.types.TerminationRegister