nessai.flows

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

Submodules

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

Package Contents

Classes

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.