distributions (pmrf.distributions)

Probability distributions.

Classes

Distribution([batch_shape, event_shape, ...])

Base class for probability distributions in NumPyro.

UniformDistribution

LogUniformDistribution

NormalDistribution

MultivariateNormalDistribution

LogNormalDistribution

JointDistribution(distributions, ...)

StackedDistribution(dists[, validate_args])

A meta-distribution that combines N independent, potentially heterogeneous scalar distributions into a single vectorized distribution of length N.

SampledDistribution([batch_shape, ...])

AnestheticDistribution(nested_samples[, ...])

Adapter for distributions represented by samples from the Anesthetic library.

FlowJAXDistribution(flow[, factory_fn, ...])

JAX-native wrapper for FlowJax's Transformed class.