experimental#

Unstable or largely untested networks, proceed with caution.

Modules

diffusion_model

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].

DiffusionModel(*args, **kwargs)

Diffusion Model as described in this overview paper [1].

FreeFormFlow(*args, **kwargs)

(IN) Implements a dimensionality-preserving Free-form Flow.