tfep.nn.transformers.sos

Sum-of-squares polynomial transformer for autoregressive normalizing flows.

Classes

SOSPolynomialTransformer([n_polynomials])	Sum-of-squares polynomial transformer module for autoregressive normalizing flows.
SOSPolynomialTransformerFunc(*args, **kwargs)	Implement the sum-of-squares polynomial transformer for triangular maps.