Index

A

ABSampler (class in azula.sample) adams_bashforth() (azula.sample.ABSampler 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.attention module azula.nn.embedding module azula.nn.layers module azula.nn.unet 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) ConvNd() (in module azula.nn.layers)

CosineSchedule (class in azula.noise) 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

EABSampler (class in azula.sample) 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) exponential_adams_bashforth() (azula.sample.EABSampler static method)

F

forward() (azula.denoise.GaussianDenoiser method) (azula.nn.attention.MultiheadSelfAttention method) (azula.nn.embedding.SineEncoding method) (azula.nn.layers.LayerNorm method) (azula.nn.layers.RMSNorm method) (azula.nn.unet.UNet method) (azula.nn.unet.UNetBlock 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_hub_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.layers) 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.PreconditionedDenoiser method)

M

MMPSDenoiser (class in azula.guidance.mmps) model_cards() (in module azula.plugins.adm) (in module azula.plugins.edm) (in module azula.plugins.eldm) (in module azula.plugins.vdm) 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.attention azula.nn.embedding azula.nn.layers azula.nn.unet azula.noise azula.plugins azula.plugins.adm azula.plugins.edm azula.plugins.eldm azula.plugins.vdm azula.sample

MultiheadSelfAttention (class in azula.nn.attention)

P

Patchify() (in module azula.nn.layers)

PCSampler (class in azula.sample) PreconditionedDenoiser (class in azula.denoise)

R

RaiseMock (class in azula.debug) RectifiedSchedule (class in azula.noise)

ReLU2 (class in azula.nn.layers) RMSNorm (class in azula.nn.layers)

S

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

set_hub_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) Unpatchify() (in module azula.nn.layers)

V

VelocityDenoiser (class in azula.plugins.vdm)

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