Samplers

[francois-rozet]

Description

This issue tracks the samplers have been requested and/or implemented in azula.sample. You are welcome to request new samplers. If you wish to contribute to Azula, this list is a good place to start.

List of requested samplers

• Annealed Langevin dynamics by Song et al. (2019)
• Denoising Diffusion Probabilistic Model (DDPM) sampler by Ho et al. (2020)
• Denoising Diffusion Implicit Model (DDIM) sampler by Song et al. (2021)
• Predictor-Corrector (PC) sampler by Song et al. (2021)
• Linear Multi-Step (LMS) sampler from crowsonkb/k-diffusion
• Diffusion Probabilistic Model Solver (DPM-Solver) sampler by Lu et al. (2022)
• DPM-Solver++ sampler by Lu et al. (2022)
• DPM-Solver-v3 sampler by Zheng et al. (2023)
• Unified Predictor-Corrector (UniPC) sampler by Zhao et al. (2023)
• Stochastic Adams (SA) sampler by Xue et al. (2023)