nessai.flows

Code related to the implementation of normalising flows and some complete implementations.

Submodules

• nessai.flows.base
• nessai.flows.distributions
• nessai.flows.fastslowflow
• nessai.flows.maf
• nessai.flows.nsf
• nessai.flows.realnvp
• nessai.flows.transforms
• nessai.flows.utils

Package Contents

Classes

BaseFlow	Base class for all normalising flows.
NFlow	Base class for flow objects implemented according to outline in nflows.
MaskedAutoregressiveFlow	Autoregressive flow with masked coupling transforms.
NeuralSplineFlow	Implementation of Neural Spline Flow
RealNVP	Implementation of RealNVP.

Functions

configure_model(config)	Setup the flow form a configuration dictionary.
reset_weights(module)	Reset parameters of a given module in place.
reset_permutations(module)	Resets permutations and linear transforms for a given module in place.