Index

A

adams_bashforth() (azula.sample.LMSSampler static method) AutoEncoder (class in azula.plugins.eldm) azula module azula.debug module azula.denoise module azula.guidance module azula.guidance.dps module azula.guidance.mmps module azula.hub module azula.linalg module azula.linalg.solve module azula.nn module

azula.nn.embedding module azula.nn.normalization module azula.nn.unet module azula.nn.utils module azula.noise module azula.plugins module azula.plugins.adm module azula.plugins.edm module azula.plugins.eldm module azula.plugins.vdm module azula.sample module

B

BetaSchedule (class in azula.plugins.adm)

C

cg() (in module azula.linalg.solve)

CrowsonSchedule (class in azula.plugins.vdm)

D

DDIMSampler (class in azula.sample) DDPMSampler (class in azula.sample)

decode() (azula.plugins.eldm.AutoEncoder method) download() (in module azula.hub) DPSSampler (class in azula.guidance.dps)

E

ElucidatedDenoiser (class in azula.plugins.edm) ElucidatedLatentDenoiser (class in azula.plugins.eldm)

ElucidatedSchedule (class in azula.plugins.edm) encode() (azula.plugins.eldm.AutoEncoder method) EulerSampler (class in azula.sample)

F

FlattenWrapper (class in azula.nn.utils) forward() (azula.denoise.GaussianDenoiser method) (azula.nn.embedding.SineEncoding method) (azula.nn.normalization.LayerNorm method) (azula.nn.unet.UNet method) (azula.nn.unet.UNetBlock method) (azula.nn.utils.FlattenWrapper method) (azula.noise.Schedule method) (azula.plugins.eldm.ElucidatedLatentDenoiser method) (azula.sample.Sampler method)

G

Gaussian (class in azula.denoise) GaussianDenoiser (class in azula.denoise)

get_dir() (in module azula.hub) gmres() (in module azula.linalg.solve)

H

HeunSampler (class in azula.sample)

I

ImprovedDenoiser (class in azula.plugins.adm)

init() (azula.sample.Sampler method)

L

LayerNorm (class in azula.nn.normalization) list_models() (in module azula.plugins.adm) (in module azula.plugins.edm) (in module azula.plugins.eldm) (in module azula.plugins.vdm) LMSSampler (class in azula.sample)

load_model() (in module azula.plugins.adm) (in module azula.plugins.edm) (in module azula.plugins.eldm) (in module azula.plugins.vdm) log_prob() (azula.denoise.Gaussian method) loss() (azula.denoise.GaussianDenoiser method)

M

MMPSDenoiser (class in azula.guidance.mmps) module azula azula.debug azula.denoise azula.guidance azula.guidance.dps azula.guidance.mmps azula.hub azula.linalg azula.linalg.solve azula.nn azula.nn.embedding azula.nn.normalization azula.nn.unet azula.nn.utils azula.noise azula.plugins azula.plugins.adm azula.plugins.edm azula.plugins.eldm azula.plugins.vdm azula.sample

P

PCSampler (class in azula.sample)

PreconditionedDenoiser (class in azula.denoise)

R

RaiseMock (class in azula.debug)

S

Sampler (class in azula.sample) Schedule (class in azula.noise)

set_dir() (in module azula.hub) SineEncoding (class in azula.nn.embedding) step() (azula.sample.Sampler method)

U

UNet (class in azula.nn.unet)

UNetBlock (class in azula.nn.unet)

V

VelocityDenoiser (class in azula.plugins.vdm)

VESchedule (class in azula.noise) VPSchedule (class in azula.noise)