tfep.nn.conditioners.made

Masked Autoregressive layer for Density Estimation (MADE) module for PyTorch.

Functions

generate_degrees(n_features[, order, ...])

Generate node degrees for MADE layers.

Classes

MADE(degrees_in, degrees_out[, ...])

An autoregressive layer implemented through masked affine layers.