Probabilistic

class pmrf.modules.Probabilistic(module: ~pmrf.modules.base.Module | None = None, distribution: ~distreqx.distributions._distribution.AbstractDistribution | None = None, target: ~typing.Callable[[~typing.Any], ~typing.Any] = <function Probabilistic.<lambda>>, constraint: ~parax.constraints.AbstractConstraint | None = None, static: ~typing.Any = None, *, model: ~pmrf.modules.base.Module | None = None)

Bases: Module, AbstractUnwrappable

A module with a probability distribution attached to a subtree.

unwrap() → Module

Evaluates and returns the underlying unwrapped PyTree node.

Returns:

The resolved underlying value, assuming no wrapped subnodes exist.

property module: Module

The probabilistic module tree.

probabilistic: Probabilize | Combine | Module