tfep.nn

Modules

conditioners	Conditioner layers for autoregressive normalizing flows.
dynamics	Dynamics layers for continuous normalizing flows.
embeddings	Embedding and encoder layers.
flows	Normalizing flow models for PyTorch.
graph	Utility functions for graphs.
masked	Masked linear transformations for PyTorch.
transformers	Transformers for autoregressive normalizing flows in PyTorch.