sample

A utility module containing helper code for sampling and resolving Dynode objects.

Functions

resolve_deterministic(obj, root_params[, ...])	Find and resolve all DeterministicParameter types.
sample_distributions(obj[, rng_key, _prefix])	Recurisvely scans data structures and samples numpyro.Distribution objects.
sample_then_resolve(parameters[, rng_key])	Copy, sample and resolve parameters, returning a jax-compliant copy.