modules (pmrf.modules)

Parameter-aware wrappers for ParamRF modules.

Classes

Module(*[, name, metadata])	Base class for parameter-aware objects in ParamRF.
Probabilistic(module, distribution, names, ...)	A joint prior over named parameters of a tree, which it wraps unchanged.
Tied(module, target, ...)	A module that ties a target field to a transformed source field.

Functions

is_module(x)	Return whether x is a Module.
validate(tree)	Validate a parameter PyTree.