coupling#

Building blocks of the CouplingFlow.

Modules

layers

Coupling layers, which transform one half of the input conditioned on the other half.

permutations

Permutations applied between the coupling layers of a CouplingFlow.

transforms

Elementwise bijections applied by the coupling layers of a CouplingFlow.

Classes

ActNorm(*args, **kwargs)

Implements an Activation Normalization (ActNorm) Layer. Activation Normalization is learned invertible normalization, using a scale (s) and a bias (b) vector::.

CouplingFlow(*args, **kwargs)

Coupling-based normalizing flow for simulation-based inference.

InvertibleLayer(*args, **kwargs)

Base class for the layers a CouplingFlow is composed of.