Environments

Environment

Abstract base class for all environments.

ScoreEnvironment

Environment with tensor samples and base model predict score \(\nabla \log p_t(x)\).

EpsilonEnvironment

Environment with tensor samples and base model predicts startpoint \(\epsilon(x, t)\).

VelocityEnvironment

Environment with tensor samples and base model predict velocity \(v(x, t)\).

EndpointEnvironment

Environment with tensor samples and base model predicts the endpoint \(x_1(x, t)\).