run

jeanspy.sampler_numpyro.NumPyroSampler.run

Class and construction: NumPyroSampler.

NumPyroSampler.run(rng_key, *args, resume='auto', save_checkpoint=True, save_samples=True, wait_for_write=False, arviz_kwargs=None, **kwargs)

Run one chunk and optionally persist it.

Notes

Inputs and units. rng_key plus model data arguments; resume is auto/True/False. save_checkpoint/save_samples/wait_for_write control persistence; arviz_kwargs are conversion options.

Returns and shape. SamplerRunResult; wait for flush/close before relying on asynchronous files.

Validity. Use an unchanged model, prior, observations and numerical configuration when resuming. Identity is checked before sampling.

Errors. ValueError for identity mismatch or invalid resume mode; FileNotFoundError for required missing state. Model, NumPyro and storage exceptions propagate.

Differentiation. Host-side sampling/storage control; this method has no physical-parameter derivative.

Parameters:

• rng_key (Any)

• args (Any)

• resume (bool | str)

• save_checkpoint (bool)

• save_samples (bool)

• wait_for_write (bool)

• arviz_kwargs (Mapping[str, Any] | None)

• kwargs (Any)

Return type:

jeanspy.sampler_numpyro.SamplerRunResult