clipppy.utils.pyro

Module Contents

clipppy.utils.pyro.init_msgr

clipppy.utils.pyro.no_grad_msgr

class clipppy.utils.pyro.NoGradMessenger

Context manager class that modifies behavior and adds side effects to stochastic functions i.e. callables containing Pyro primitive statements.

This is the base Messenger class. It implements the default behavior for all Pyro primitives, so that the joint distribution induced by a stochastic function fn is identical to the joint distribution induced by Messenger()(fn).

Class of transformers for messages passed during inference. Most inference operations are implemented in subclasses of this.

static _pyro_post_param(msg: clipppy.utils.typing._Site)

clipppy.utils.pyro.depoutine(obj: Union[pyro.poutine.messenger._bound_partial, Any], msgr_type: Type[pyro.poutine.messenger.Messenger] = ConditionMessenger)

clipppy.utils.pyro.init_fn(site: clipppy.utils.typing._Site) → torch.Tensor