Index A | C | D | E | F | G | I | L | M | N | P | S | T | U | V | X A alpha (diffusionlab.dynamics.DiffusionProcess attribute) (diffusionlab.dynamics.FlowMatchingProcess attribute) (diffusionlab.dynamics.VarianceExplodingProcess attribute) (diffusionlab.dynamics.VariancePreservingProcess attribute) alpha_prime (diffusionlab.dynamics.DiffusionProcess attribute) (diffusionlab.dynamics.FlowMatchingProcess attribute) (diffusionlab.dynamics.VarianceExplodingProcess attribute) (diffusionlab.dynamics.VariancePreservingProcess attribute) C convert_vector_field_type() (in module diffusionlab.vector_fields) create_gmm_vector_field_fns() (in module diffusionlab.distributions.gmm.utils) D DDMSampler (class in diffusionlab.samplers) diffusion_process (diffusionlab.losses.DiffusionLoss attribute) (diffusionlab.samplers.DDMSampler attribute) (diffusionlab.samplers.EulerMaruyamaSampler attribute) (diffusionlab.samplers.Sampler attribute) diffusionlab module diffusionlab.distributions module diffusionlab.distributions.base module diffusionlab.distributions.empirical module diffusionlab.distributions.gmm module diffusionlab.distributions.gmm.gmm module diffusionlab.distributions.gmm.iso_gmm module diffusionlab.distributions.gmm.iso_hom_gmm module diffusionlab.distributions.gmm.low_rank_gmm module diffusionlab.distributions.gmm.utils module diffusionlab.dynamics module diffusionlab.losses module diffusionlab.samplers module diffusionlab.schedulers module diffusionlab.vector_fields module DiffusionLoss (class in diffusionlab.losses) DiffusionProcess (class in diffusionlab.dynamics) dist_hparams (diffusionlab.distributions.base.Distribution attribute) (diffusionlab.distributions.empirical.EmpiricalDistribution attribute) (diffusionlab.distributions.gmm.gmm.GMM attribute) (diffusionlab.distributions.gmm.iso_gmm.IsoGMM attribute) (diffusionlab.distributions.gmm.iso_hom_gmm.IsoHomGMM attribute) (diffusionlab.distributions.gmm.low_rank_gmm.LowRankGMM attribute) dist_params (diffusionlab.distributions.base.Distribution attribute) (diffusionlab.distributions.empirical.EmpiricalDistribution attribute) (diffusionlab.distributions.gmm.gmm.GMM attribute) (diffusionlab.distributions.gmm.iso_gmm.IsoGMM attribute) (diffusionlab.distributions.gmm.iso_hom_gmm.IsoHomGMM attribute) (diffusionlab.distributions.gmm.low_rank_gmm.LowRankGMM attribute) Distribution (class in diffusionlab.distributions.base) E EmpiricalDistribution (class in diffusionlab.distributions.empirical) eps() (diffusionlab.distributions.base.Distribution method) (diffusionlab.distributions.empirical.EmpiricalDistribution method) (diffusionlab.distributions.gmm.gmm.GMM method) (diffusionlab.distributions.gmm.iso_gmm.IsoGMM method) (diffusionlab.distributions.gmm.iso_hom_gmm.IsoHomGMM method) (diffusionlab.distributions.gmm.low_rank_gmm.LowRankGMM method) EulerMaruyamaSampler (class in diffusionlab.samplers) F FlowMatchingProcess (class in diffusionlab.dynamics) forward() (diffusionlab.dynamics.DiffusionProcess method) G get_sample_step_function() (diffusionlab.samplers.DDMSampler method) (diffusionlab.samplers.EulerMaruyamaSampler method) (diffusionlab.samplers.Sampler method) get_ts() (diffusionlab.schedulers.Scheduler method) (diffusionlab.schedulers.UniformScheduler method) get_vector_field() (diffusionlab.distributions.base.Distribution method) GMM (class in diffusionlab.distributions.gmm.gmm) gmm_x0() (in module diffusionlab.distributions.gmm.gmm) I iso_gmm_x0() (in module diffusionlab.distributions.gmm.iso_gmm) iso_hom_gmm_x0() (in module diffusionlab.distributions.gmm.iso_hom_gmm) IsoGMM (class in diffusionlab.distributions.gmm.iso_gmm) IsoHomGMM (class in diffusionlab.distributions.gmm.iso_hom_gmm) L loss() (diffusionlab.losses.DiffusionLoss method) low_rank_gmm_x0() (in module diffusionlab.distributions.gmm.low_rank_gmm) LowRankGMM (class in diffusionlab.distributions.gmm.low_rank_gmm) M module diffusionlab diffusionlab.distributions diffusionlab.distributions.base diffusionlab.distributions.empirical diffusionlab.distributions.gmm diffusionlab.distributions.gmm.gmm diffusionlab.distributions.gmm.iso_gmm diffusionlab.distributions.gmm.iso_hom_gmm diffusionlab.distributions.gmm.low_rank_gmm diffusionlab.distributions.gmm.utils diffusionlab.dynamics diffusionlab.losses diffusionlab.samplers diffusionlab.schedulers diffusionlab.vector_fields N num_noise_draws_per_sample (diffusionlab.losses.DiffusionLoss attribute) P prediction_loss() (diffusionlab.losses.DiffusionLoss method) S sample() (diffusionlab.distributions.base.Distribution method) (diffusionlab.distributions.empirical.EmpiricalDistribution method) (diffusionlab.distributions.gmm.gmm.GMM method) (diffusionlab.distributions.gmm.iso_gmm.IsoGMM method) (diffusionlab.distributions.gmm.iso_hom_gmm.IsoHomGMM method) (diffusionlab.distributions.gmm.low_rank_gmm.LowRankGMM method) (diffusionlab.samplers.Sampler method) sample_step (diffusionlab.samplers.DDMSampler attribute) (diffusionlab.samplers.EulerMaruyamaSampler attribute) (diffusionlab.samplers.Sampler attribute) sample_trajectory() (diffusionlab.samplers.Sampler method) Sampler (class in diffusionlab.samplers) Scheduler (class in diffusionlab.schedulers) score() (diffusionlab.distributions.base.Distribution method) (diffusionlab.distributions.empirical.EmpiricalDistribution method) (diffusionlab.distributions.gmm.gmm.GMM method) (diffusionlab.distributions.gmm.iso_gmm.IsoGMM method) (diffusionlab.distributions.gmm.iso_hom_gmm.IsoHomGMM method) (diffusionlab.distributions.gmm.low_rank_gmm.LowRankGMM method) sigma (diffusionlab.dynamics.DiffusionProcess attribute) (diffusionlab.dynamics.FlowMatchingProcess attribute) (diffusionlab.dynamics.VarianceExplodingProcess attribute) (diffusionlab.dynamics.VariancePreservingProcess attribute) sigma_prime (diffusionlab.dynamics.DiffusionProcess attribute) (diffusionlab.dynamics.FlowMatchingProcess attribute) (diffusionlab.dynamics.VarianceExplodingProcess attribute) (diffusionlab.dynamics.VariancePreservingProcess attribute) T target (diffusionlab.losses.DiffusionLoss attribute) U UniformScheduler (class in diffusionlab.schedulers) use_stochastic_sampler (diffusionlab.samplers.DDMSampler attribute) (diffusionlab.samplers.EulerMaruyamaSampler attribute) (diffusionlab.samplers.Sampler attribute) V v() (diffusionlab.distributions.base.Distribution method) (diffusionlab.distributions.empirical.EmpiricalDistribution method) (diffusionlab.distributions.gmm.gmm.GMM method) (diffusionlab.distributions.gmm.iso_gmm.IsoGMM method) (diffusionlab.distributions.gmm.iso_hom_gmm.IsoHomGMM method) (diffusionlab.distributions.gmm.low_rank_gmm.LowRankGMM method) VarianceExplodingProcess (class in diffusionlab.dynamics) VariancePreservingProcess (class in diffusionlab.dynamics) vector_field (diffusionlab.samplers.DDMSampler attribute) (diffusionlab.samplers.EulerMaruyamaSampler attribute) (diffusionlab.samplers.Sampler attribute) vector_field_type (diffusionlab.losses.DiffusionLoss attribute) (diffusionlab.samplers.DDMSampler attribute) (diffusionlab.samplers.EulerMaruyamaSampler attribute) (diffusionlab.samplers.Sampler attribute) VectorFieldType (class in diffusionlab.vector_fields) X x0() (diffusionlab.distributions.base.Distribution method) (diffusionlab.distributions.empirical.EmpiricalDistribution method) (diffusionlab.distributions.gmm.gmm.GMM method) (diffusionlab.distributions.gmm.iso_gmm.IsoGMM method) (diffusionlab.distributions.gmm.iso_hom_gmm.IsoHomGMM method) (diffusionlab.distributions.gmm.low_rank_gmm.LowRankGMM method)