uniform_sample

jaxns.nested_samplers.common.uniform_sample

Module Contents

draw_uniform_samples(keys, model, method='vmap')[source]

Get initial live points from uniformly sampling the entire prior.

Parameters:

keys (jaxns.internals.types.PRNGKey) – PRNGKey
model (jaxns.framework.bases.BaseAbstractModel) – the model
method (str) – which way to draw the init points. vmap is vectorised, and for performant but uses more memory.

Returns:

uniformly drawn samples within -inf bound

Return type:

jaxns.nested_samplers.common.types.Sample