Bases: object
A bundle of sampled configurations and their reference log-probabilities.
-
samples[source]
Physical configurations, shape (n_samples, N).
-
log_probs[source]
Reference log-density log p_ref(sigma) at each sample,
shape (n_samples,).
The arrays are kept as sharded JAX arrays. Use save() /
load() to round-trip through a lightweight .npz file.
- Methods
-
classmethod load(path)[source]
Load a bundle written by save().
Both arrays are sharded along the sample axis "S" so they are ready
for use in a distributed ISState.
- Return type:
SamplesWithProb
- Parameters:
path (str)
-
save(path)[source]
Save the bundle to a .npz file at path.
Sharded arrays are gathered to fully-addressable host arrays; only the
main process touches the disk.
- Return type:
None
- Parameters:
path (str)
