modules (pmrf.modules)

Parameter-aware wrappers for ParamRF modules.

Classes

Module(*[, name, metadata])

Base class for parameter-aware objects in ParamRF.

Probabilistic(module, distribution, target, ...)

A module with a probability distribution attached to a subtree.

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.