Index

_

__call__() (azula.guidance.tds.TDSSampler method) (azula.noise.Schedule method) (azula.sample.Sampler method)

A

AblatedDenoiser (class in azula.plugins.adm) AutoEncoder (class in azula.plugins.eldm) (class in azula.plugins.flux) (class in azula.plugins.sana) (class in azula.plugins.sd) azula module azula.debug module azula.denoise module azula.guidance module azula.guidance.cfg module azula.guidance.diffpir module azula.guidance.dps module azula.guidance.jfps module azula.guidance.mmps module azula.guidance.pgdm module azula.guidance.repaint module azula.guidance.tds module azula.guidance.tmpd module azula.hub module azula.linalg module azula.linalg.covariance 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.nn.utils module azula.nn.vit module azula.noise module azula.plugins module azula.plugins.adm module azula.plugins.edm module azula.plugins.eldm module azula.plugins.flux module azula.plugins.jit module azula.plugins.sana module azula.plugins.sd module azula.plugins.utils module azula.sample module

C

CFGDenoiser (class in azula.guidance.cfg) cg() (in module azula.linalg.solve) checkpoint() (in module azula.nn.utils)

ConvNd() (in module azula.nn.layers) CosineSchedule (class in azula.noise) Covariance (class in azula.linalg.covariance)

D

DDIMSampler (class in azula.sample) DDPMSampler (class in azula.sample) DecaySchedule (class in azula.noise) decode() (azula.plugins.eldm.AutoEncoder method) (azula.plugins.flux.AutoEncoder method) (azula.plugins.sana.AutoEncoder method) (azula.plugins.sd.AutoEncoder method)

Denoiser (class in azula.denoise) DiagonalCovariance (class in azula.linalg.covariance) DiffPIRDenoiser (class in azula.guidance.diffpir) DiracPosterior (class in azula.denoise) download() (in module azula.hub) DPLRCovariance (class in azula.linalg.covariance) 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) (azula.plugins.flux.AutoEncoder method) (azula.plugins.sana.AutoEncoder method) (azula.plugins.sd.AutoEncoder method)

EulerSampler (class in azula.sample)

F

FluxDenoiser (class in azula.plugins.flux) forward() (azula.denoise.Denoiser method) (azula.denoise.GaussianDenoiser method) (azula.denoise.KarrasDenoiser method) (azula.denoise.SimpleDenoiser method) (azula.guidance.cfg.CFGDenoiser method) (azula.nn.attention.MultiheadSelfAttention method) (azula.nn.embedding.SineEncoding method) (azula.nn.layers.LayerNorm method) (azula.nn.layers.RMSNorm method) (azula.nn.layers.SwiGLU method) (azula.nn.unet.UNet method) (azula.nn.unet.UNetBlock method) (azula.nn.vit.ViT method) (azula.nn.vit.ViTBlock method) (azula.plugins.adm.AblatedDenoiser method) (azula.plugins.edm.ElucidatedDenoiser method) (azula.plugins.eldm.ElucidatedLatentDenoiser method) (azula.plugins.flux.FluxDenoiser method) (azula.plugins.flux.TextEncoder method) (azula.plugins.jit.JITDenoiser method) (azula.plugins.sana.SanaDenoiser method) (azula.plugins.sana.TextEncoder method) (azula.plugins.sd.StableDenoiser method) (azula.plugins.sd.TextEncoder method)

FullCovariance (class in azula.linalg.covariance)

G

GaussianDenoiser (class in azula.denoise) GaussianPosterior (class in azula.denoise) get_hub_dir() (in module azula.hub)

get_module_device() (in module azula.nn.utils) get_module_dtype() (in module azula.nn.utils) gmres() (in module azula.linalg.solve)

H

HeunSampler (class in azula.sample)

I

init() (azula.sample.Sampler method)

IsotropicCovariance (class in azula.linalg.covariance)

J

JFPSDenoiser (class in azula.guidance.jfps)

JITDenoiser (class in azula.plugins.jit)

K

KarrasDenoiser (class in azula.denoise)

KroneckerCovariance (class in azula.linalg.covariance)

L

LayerNorm (class in azula.nn.layers) load_cards() (in module azula.plugins.utils) load_model() (in module azula.plugins.adm) (in module azula.plugins.edm) (in module azula.plugins.eldm) (in module azula.plugins.flux) (in module azula.plugins.jit) (in module azula.plugins.sana) (in module azula.plugins.sd)

log_prob() (azula.denoise.GaussianPosterior method) loss() (azula.denoise.KarrasDenoiser method) (azula.denoise.SimpleDenoiser method)

M

MMPSDenoiser (class in azula.guidance.mmps) module azula azula.debug azula.denoise azula.guidance azula.guidance.cfg azula.guidance.diffpir azula.guidance.dps azula.guidance.jfps azula.guidance.mmps azula.guidance.pgdm azula.guidance.repaint azula.guidance.tds azula.guidance.tmpd azula.hub azula.linalg azula.linalg.covariance azula.linalg.solve azula.nn azula.nn.attention azula.nn.embedding azula.nn.layers azula.nn.unet azula.nn.utils azula.nn.vit azula.noise azula.plugins azula.plugins.adm azula.plugins.edm azula.plugins.eldm azula.plugins.flux azula.plugins.jit azula.plugins.sana azula.plugins.sd azula.plugins.utils azula.sample

MultiheadSelfAttention (class in azula.nn.attention)

P

Patchify() (in module azula.nn.layers) PCSampler (class in azula.sample)

PGDMSampler (class in azula.guidance.pgdm) Posterior (class in azula.denoise) promote_dtype() (in module azula.nn.utils)

R

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

ReLU2 (class in azula.nn.layers) RePaintSampler (class in azula.guidance.repaint) RMSNorm (class in azula.nn.layers)

S

Sampler (class in azula.sample) SanaDenoiser (class in azula.plugins.sana) Schedule (class in azula.noise) set_hub_dir() (in module azula.hub) SimpleDenoiser (class in azula.denoise)

SineEncoding (class in azula.nn.embedding) skip_init (class in azula.nn.utils) StableDenoiser (class in azula.plugins.sd) step() (azula.sample.Sampler method) SwiGLU (class in azula.nn.layers)

T

TDSSampler (class in azula.guidance.tds) TextEncoder (class in azula.plugins.flux) (class in azula.plugins.sana) (class in azula.plugins.sd)

TMPDenoiser (class in azula.guidance.tmpd)

U

UNet (class in azula.nn.unet)

UNetBlock (class in azula.nn.unet) Unpatchify() (in module azula.nn.layers)

V

vABSampler (class in azula.sample) VESchedule (class in azula.noise)

ViT (class in azula.nn.vit) ViTBlock (class in azula.nn.vit) VPSchedule (class in azula.noise)

X

xEABSampler (class in azula.sample)

Z

zABSampler (class in azula.sample)

zEABSampler (class in azula.sample)