diffusion_model

Functions

find_noise_schedule()

Classes

CosineNoiseSchedule([min_log_snr, ...])	Cosine noise schedule for diffusion models.
DiffusionModel(*args, **kwargs)	Score-based diffusion model for simulation-based inference as described in [1]:
EDMNoiseSchedule([sigma_data, sigma_min, ...])	EDM noise schedule for diffusion models.
NoiseSchedule(name, variance_type[, weighting])	Noise schedule for diffusion models.