experimental

Unstable or largely untested networks, proceed with caution.

Functions

DiffusionModel(*args, **kwargs)

Classes

CIF(*args, **kwargs)	(IN) Implements a continuously indexed flow (CIF) with a CouplingFlow bijection and ConditionalGaussian distributions p and q.
ContinuousTimeConsistencyModel(*args, **kwargs)	(IN) Implements an sCM (simple, stable, and scalable Consistency Model) with continous-time Consistency Training (CT) as described in [1].
FreeFormFlow(*args, **kwargs)	(IN) Implements a dimensionality-preserving Free-form Flow.