Index A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | W | X | Y A adjoint() (diffuse.base_forward_model.ForwardModel method) apply() (diffuse.base_forward_model.ForwardModel method) attention() (diffuse.neural_network.blocks.attention.AttnBlock method) AttnBlock (class in diffuse.neural_network.blocks.attention) auxiliary (diffuse.denoisers.pnp_denoiser.PnPDenoiserState attribute) B BaseDenoiser (class in diffuse.denoisers.base) batch_step() (diffuse.denoisers.pnp_denoiser.PnPDenoiser method) C ChurnedIntegrator (class in diffuse.integrator) CondUNet2D (class in diffuse.neural_network.nn.condUNet) CondUNet2DOutput (class in diffuse.neural_network.nn.params) CosineSchedule (class in diffuse.diffusion.sde) D DAPSDenoiser (class in diffuse.denoisers.cond.daps) DDIMIntegrator (class in diffuse.integrator) DDIMTimer (class in diffuse.timer.base) decode() (diffuse.neural_network.nn.sdVae.SDVae method) Decoder (class in diffuse.neural_network.blocks.decoder) Denoiser (class in diffuse.denoisers.denoiser), [1] DenoiserState (class in diffuse.denoisers.base) DiffPIRDenoiser (class in diffuse.denoisers.cond.diffpir) diffuse.denoisers.base module diffuse.denoisers.denoiser module diffuse.denoisers.pnp_denoiser module diffuse.denoisers.utils module diffuse.neural_network.blocks.attention module diffuse.neural_network.blocks.decoder module diffuse.neural_network.blocks.downsample module diffuse.neural_network.blocks.encoder module diffuse.neural_network.blocks.resnet_block module diffuse.neural_network.blocks.time_embedding module diffuse.neural_network.blocks.timestep module diffuse.neural_network.blocks.upsample module diffuse.neural_network.nn.params module diffuse.utils.logger module diffuse.utils.mapping module Downsample (class in diffuse.neural_network.blocks.downsample) DPMpp2sIntegrator (class in diffuse.integrator) DPSDenoiser (class in diffuse.denoisers.cond.dps) DPSGSGDenoiser (class in diffuse.denoisers.cond.dps_gsg) E EDM (class in diffuse.diffusion.sde) encode() (diffuse.neural_network.nn.sdVae.SDVae method) Encoder (class in diffuse.neural_network.blocks.encoder) EnKGDenoiser (class in diffuse.denoisers.cond.enkg) ess() (in module diffuse.denoisers.utils) ess_high (diffuse.denoisers.cond.fps.FPSDenoiser attribute) ess_low (diffuse.denoisers.cond.fps.FPSDenoiser attribute) EulerIntegrator (class in diffuse.integrator) EulerMaruyamaIntegrator (class in diffuse.integrator) F Flow (class in diffuse.diffusion.sde) forward_model (diffuse.denoisers.pnp_denoiser.PnPDenoiser attribute) ForwardModel (class in diffuse.base_forward_model) FPSDenoiser (class in diffuse.denoisers.cond.fps) from_tuple() (diffuse.neural_network.nn.params.CondUNet2DOutput method) (diffuse.neural_network.nn.params.SDVaeOutput method) G generate() (diffuse.denoisers.base.BaseDenoiser method) (diffuse.denoisers.cond.enkg.EnKGDenoiser method) (diffuse.denoisers.denoiser.Denoiser method), [1] get_sinusoidal_embedding() (in module diffuse.neural_network.blocks.time_embedding) H HeunIntegrator (class in diffuse.integrator) HeunTimer (class in diffuse.timer.base) I init() (diffuse.denoisers.base.BaseDenoiser method) (diffuse.denoisers.denoiser.Denoiser method), [1] (diffuse.denoisers.pnp_denoiser.PnPDenoiser method) (diffuse.integrator.Integrator method) integrate() (diffuse.diffusion.sde.CosineSchedule method) (diffuse.diffusion.sde.LinearSchedule method) Integrator (class in diffuse.integrator) integrator (diffuse.denoisers.denoiser.Denoiser attribute) (diffuse.denoisers.pnp_denoiser.PnPDenoiser attribute) integrator_state (diffuse.denoisers.base.DenoiserState attribute) IntegratorState (class in diffuse.integrator) items() (diffuse.neural_network.nn.params.CondUNet2DOutput method) (diffuse.neural_network.nn.params.SDVaeOutput method) K keys() (diffuse.neural_network.nn.params.CondUNet2DOutput method) (diffuse.neural_network.nn.params.SDVaeOutput method) L langevin_sampling() (diffuse.denoisers.cond.pnpdm.PnPDMDenoiser method) LinearSchedule (class in diffuse.diffusion.sde) log_ess() (in module diffuse.denoisers.utils) log_weights (diffuse.denoisers.pnp_denoiser.PnPDenoiserState attribute) logvar (diffuse.neural_network.nn.params.SDVaeOutput attribute), [1] M make_in_axes_except() (in module diffuse.utils.mapping) mask_history (diffuse.base_forward_model.MeasurementState attribute) mean (diffuse.neural_network.nn.params.SDVaeOutput attribute), [1] MeasurementState (class in diffuse.base_forward_model) model (diffuse.denoisers.denoiser.Denoiser attribute) module diffuse.denoisers.base diffuse.denoisers.denoiser diffuse.denoisers.pnp_denoiser diffuse.denoisers.utils diffuse.neural_network.blocks.attention diffuse.neural_network.blocks.decoder diffuse.neural_network.blocks.downsample diffuse.neural_network.blocks.encoder diffuse.neural_network.blocks.resnet_block diffuse.neural_network.blocks.time_embedding diffuse.neural_network.blocks.timestep diffuse.neural_network.blocks.upsample diffuse.neural_network.nn.params diffuse.utils.logger diffuse.utils.mapping N noise() (diffuse.predictor.Predictor method) noise_level() (diffuse.diffusion.sde.EDM method) (diffuse.diffusion.sde.Flow method) (diffuse.diffusion.sde.SDE method) normalize_log_weights() (in module diffuse.denoisers.utils) O output (diffuse.neural_network.nn.params.CondUNet2DOutput attribute), [1] (diffuse.neural_network.nn.params.SDVaeOutput attribute), [1] P PiGDMDenoiser (class in diffuse.denoisers.cond.pigdm) PixelShuffle (class in diffuse.neural_network.blocks.upsample) pmap_reshaping() (in module diffuse.utils.mapping) pmap_unshaping() (in module diffuse.utils.mapping) pmapper() (in module diffuse.utils.mapping) PnPDenoiser (class in diffuse.denoisers.pnp_denoiser) PnPDenoiserState (class in diffuse.denoisers.pnp_denoiser) PnPDMDenoiser (class in diffuse.denoisers.cond.pnpdm) pooled_posterior_logpdf() (diffuse.denoisers.pnp_denoiser.PnPDenoiser method) position (diffuse.denoisers.pnp_denoiser.PnPDenoiserState attribute) (diffuse.integrator.IntegratorState attribute) posterior_logpdf() (diffuse.denoisers.pnp_denoiser.PnPDenoiser method) Predictor (class in diffuse.predictor) predictor (diffuse.denoisers.denoiser.Denoiser attribute) R replace() (diffuse.neural_network.nn.params.CondUNet2DOutput method) (diffuse.neural_network.nn.params.SDVaeOutput method) resample (diffuse.denoisers.cond.fps.FPSDenoiser attribute) resample_particles() (in module diffuse.denoisers.utils) resampler() (diffuse.denoisers.cond.fps.FPSDenoiser method) ResnetBlock (class in diffuse.neural_network.blocks.resnet_block) restore() (diffuse.base_forward_model.ForwardModel method) reverse_diffuse() (diffuse.denoisers.cond.pnpdm.PnPDMDenoiser method) rng_key (diffuse.integrator.IntegratorState attribute) S score (diffuse.denoisers.pnp_denoiser.PnPDenoiser attribute) score() (diffuse.predictor.Predictor method) SDE (class in diffuse.diffusion.sde) sde (diffuse.denoisers.pnp_denoiser.PnPDenoiser attribute) sde_coefficients() (diffuse.diffusion.sde.EDM method) (diffuse.diffusion.sde.Flow method) (diffuse.diffusion.sde.SDE method) SDVae (class in diffuse.neural_network.nn.sdVae) SDVaeOutput (class in diffuse.neural_network.nn.params) signal_level() (diffuse.diffusion.sde.EDM method) (diffuse.diffusion.sde.Flow method) (diffuse.diffusion.sde.SDE method) std (diffuse.base_forward_model.ForwardModel attribute) step (diffuse.integrator.IntegratorState attribute) step() (diffuse.denoisers.base.BaseDenoiser method) (diffuse.denoisers.cond.daps.DAPSDenoiser method) (diffuse.denoisers.cond.diffpir.DiffPIRDenoiser method) (diffuse.denoisers.cond.dps.DPSDenoiser method) (diffuse.denoisers.cond.dps_gsg.DPSGSGDenoiser method) (diffuse.denoisers.cond.enkg.EnKGDenoiser method) (diffuse.denoisers.cond.fps.FPSDenoiser method) (diffuse.denoisers.cond.pigdm.PiGDMDenoiser method) (diffuse.denoisers.cond.pnpdm.PnPDMDenoiser method) (diffuse.denoisers.cond.tmp.TMPDenoiser method) (diffuse.denoisers.denoiser.Denoiser method), [1] (diffuse.denoisers.pnp_denoiser.PnPDenoiser method) stratified_resampling() (in module diffuse.denoisers.utils) T Timer (class in diffuse.timer.base) TimestepBlock (class in diffuse.neural_network.blocks.timestep) TimestepEmbedding (class in diffuse.neural_network.blocks.time_embedding) TimestepEmbedSequential (class in diffuse.neural_network.blocks.timestep) Timesteps (class in diffuse.neural_network.blocks.time_embedding) TMPDenoiser (class in diffuse.denoisers.cond.tmp) to_tuple() (diffuse.neural_network.nn.params.CondUNet2DOutput method) (diffuse.neural_network.nn.params.SDVaeOutput method) U Upsample (class in diffuse.neural_network.blocks.upsample) V values() (diffuse.neural_network.nn.params.CondUNet2DOutput method) (diffuse.neural_network.nn.params.SDVaeOutput method) velocity() (diffuse.predictor.Predictor method) VpTimer (class in diffuse.timer.base) W weights_tweedie() (in module diffuse.denoisers.utils) X x0() (diffuse.predictor.Predictor method) x0_shape (diffuse.denoisers.denoiser.Denoiser attribute) Y y (diffuse.base_forward_model.MeasurementState attribute) y_noiser() (diffuse.denoisers.cond.fps.FPSDenoiser method) (diffuse.denoisers.pnp_denoiser.PnPDenoiser method)
