antsnormflows.nets.made

Implementation of MADE. Code taken from https://github.com/bayesiains/nsf

Classes

MADE(features, hidden_features[, ...])	Implementation of MADE.
MaskedFeedforwardBlock(in_degrees, ...[, ...])	A feedforward block based on a masked linear module.
MaskedLinear(in_degrees, out_features, ...)	A linear module with a masked weight matrix.
MaskedResidualBlock(in_degrees, ...[, ...])	A residual block containing masked linear modules.