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.dit 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.plugins.vdm 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) DiT (class in azula.nn.dit) DiTBlock (class in azula.nn.dit) DMLRCovariance (class in azula.linalg.covariance) 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.dit.DiT method) (azula.nn.dit.DiTBlock method) (azula.nn.layers.LayerNorm method) (azula.nn.layers.RMSNorm method) (azula.nn.layers.SineEncoding method) (azula.nn.layers.SwiGLU method) (azula.nn.unet.UNet method) (azula.nn.unet.UNetBlock method) (azula.nn.vit.ViT 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)

from_data() (azula.linalg.covariance.DPLRCovariance static 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) ItoSampler (class in azula.sample)

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) (in module azula.plugins.vdm)

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.dit 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.plugins.vdm 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.layers) 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) VelocityDenoiser (class in azula.plugins.vdm)

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