tfep.nn.conditioners

Conditioner layers for autoregressive normalizing flows.

Modules

conditioner

Base conditioner classes for autoregressive flows.

made

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