tfep.nn.transformers.transformer

Base transformer classes for autoregressive flows.

Classes

MAFTransformer(*args, **kwargs)	A transformer for a masked autoregressive flow (MAF).
Transformer(*args, **kwargs)	A transformer for an autoregressive flow.