tfep.nn.embeddings.mafembed

Embedding layers for masked autoregressive flows (MAF).

Classes

FlipInvariantEmbedding(n_features_in, ...[, ...])	Embeds vector features into a representation invariant to sign flips.
MAFEmbedding(*args, **kwargs)	An embedding layer for a masked autoregressive flow.
MixedEmbedding(n_features_in, ...)	Utility class mixing multiple embeddings.
PeriodicEmbedding(n_features_in, limits[, ...])	Lift periodic degrees of freedom into a periodic representation (cos, sin).