experimental#

Unstable or largely untested networks, proceed with caution.

Classes

CIF(*args, **kwargs)

Implements a continuously indexed flow (CIF) with a CouplingFlow bijection and ConditionalGaussian distributions p and q.

ContinuousTimeConsistencyModel(*args, **kwargs)

Implements an sCM (simple, stable, and scalable Consistency Model) with continous-time Consistency Training (CT) as described in [1].

FreeFormFlow(*args, **kwargs)

Implements a dimensionality-preserving Free-form Flow.