Index

A

alpha() (nami.core.typing.NoiseSchedule method) (nami.schedules.base.NoiseSchedule method) (nami.schedules.edm.EDMSchedule method) (nami.schedules.ve.VESchedule method) (nami.schedules.vp.VPSchedule method)

arg_constraints (nami.distributions.normal.DiagonalNormal attribute) (nami.distributions.normal.StandardNormal attribute) as_tuple() (in module nami.core.specs)

B

base (nami.processes.fm.FlowMatchingProcess property) batch_shape (nami.processes.diffusion.DiffusionProcess property) (nami.processes.fm.FlowMatchingProcess property) bkg (nami_toys.gaussian.GaussianMixture property) (nami_toys.parameterised.ParameterisedGaussian property) bkg_cov (nami_toys.gaussian.GaussianMixture attribute) (nami_toys.parameterised.ParameterisedGaussian attribute)

bkg_loc (nami_toys.gaussian.GaussianMixture attribute) (nami_toys.parameterised.ParameterisedGaussian attribute) bkg_scale (nami_toys.shell.GaussianShell attribute) bound (nami_toys.checkerboard.Checkerboard attribute) broadcast() (in module nami.core.broadcast) BrownianGamma (class in nami.interpolants.gamma)

C

call_and_divergence() (nami.fields.base.VectorField method) cells (nami_toys.checkerboard.Checkerboard attribute)

Checkerboard (class in nami_toys.checkerboard) CosinePath (class in nami.paths.cosine)

D

d (nami_toys.gaussian.GaussianMixture property) (nami_toys.parameterised.ParameterisedGaussian property) describe() (in module nami.debug) DiagonalNormal (class in nami.distributions.normal) Diffusion (class in nami.processes.diffusion) diffusion() (nami.core.typing.NoiseSchedule method) (nami.schedules.base.NoiseSchedule method) (nami.schedules.edm.EDMSchedule method) (nami.schedules.ve.VESchedule method) (nami.schedules.vp.VPSchedule method)

DiffusionProcess (class in nami.processes.diffusion) divergence_stats() (in module nami.diagnostics) DivergenceEstimator (class in nami.core.typing) (class in nami.divergence.base) drift() (nami.core.typing.NoiseSchedule method) (nami.schedules.base.NoiseSchedule method) (nami.schedules.edm.EDMSchedule method) (nami.schedules.ve.VESchedule method) (nami.schedules.vp.VPSchedule method) DriftFromVelocityScore (class in nami.interpolants.transforms) dtype (nami.core.specs.TensorSpec attribute)

E

EDMSchedule (class in nami.schedules.edm) eps (nami.interpolants.gamma.BrownianGamma attribute) (nami.interpolants.gamma.ScaledBrownianGamma attribute) (nami.interpolants.transforms.ScoreFromNoise attribute) eps_to_score() (in module nami.fields.diffusion) eps_to_x0() (in module nami.fields.diffusion) eta_model (nami.interpolants.transforms.ScoreFromNoise attribute) EulerMaruyama (class in nami.solvers.sde) event_ndim (nami.core.specs.TensorSpec property) (nami.fields.base.VectorField property) (nami.fields.transformer.TransformerVelocityField property) (nami.fields.velocity.VelocityField property) (nami.interpolants.transforms.DriftFromVelocityScore property) (nami.interpolants.transforms.MarkovizationDriftFromVelocityScore property) (nami.interpolants.transforms.MirrorVelocityFromScore property) (nami.interpolants.transforms.ScoreFromNoise property) (nami.lazy.LazyField property) (nami.lazy.UnconditionalField property)

event_numel() (in module nami.core.specs) event_shape (nami.core.specs.TensorSpec attribute) (nami.processes.diffusion.DiffusionProcess property) (nami.processes.fm.FlowMatchingProcess property) ExactDivergence (class in nami.divergence.exact) expand() (nami.distributions.normal.DiagonalNormal method) (nami.distributions.normal.StandardNormal method) expand_distribution() (in module nami.distributions.base)

F

field (nami.processes.fm.FlowMatchingProcess property) field_stats() (in module nami.diagnostics) fit() (nami_toys.standardise.Standardiser class method) flatten_event() (in module nami.core.specs) FlowMatching (class in nami.processes.fm) FlowMatchingProcess (class in nami.processes.fm) fm_loss() (in module nami.losses.fm) forward() (nami.components.mlp.MLPBackbone method) (nami.components.time.ScalarTimeEmbedding method) (nami.components.time.SinusoidalTimeEmbedding method) (nami.components.transformer.TransformerBackbone method) (nami.components.transformer.TransformerBlock method) (nami.fields.base.VectorField method) (nami.fields.transformer.TransformerVelocityField method) (nami.fields.velocity.VelocityField method) (nami.interpolants.transforms.DriftFromVelocityScore method) (nami.interpolants.transforms.MarkovizationDriftFromVelocityScore method) (nami.interpolants.transforms.MirrorVelocityFromScore method) (nami.interpolants.transforms.ScoreFromNoise method) (nami.lazy.LazyDistribution method) (nami.lazy.LazyField method) (nami.lazy.UnconditionalDistribution method) (nami.lazy.UnconditionalField method) (nami.processes.diffusion.Diffusion method) (nami.processes.fm.FlowMatching method)

from_sigma() (nami.interpolants.gamma.ScaledBrownianGamma class method)

G

gamma() (nami.interpolants.gamma.BrownianGamma method) (nami.interpolants.gamma.GammaSchedule method) (nami.interpolants.gamma.ScaledBrownianGamma method) (nami.interpolants.gamma.ZeroGamma method) gamma_dot() (nami.interpolants.gamma.BrownianGamma method) (nami.interpolants.gamma.GammaSchedule method) (nami.interpolants.gamma.ScaledBrownianGamma method) (nami.interpolants.gamma.ZeroGamma method) gamma_gamma_dot() (nami.interpolants.gamma.BrownianGamma method) (nami.interpolants.gamma.GammaSchedule method) (nami.interpolants.gamma.ScaledBrownianGamma method) (nami.interpolants.gamma.ZeroGamma method) gamma_schedule (nami.interpolants.transforms.DriftFromVelocityScore attribute) (nami.interpolants.transforms.ScoreFromNoise attribute)

GammaSchedule (class in nami.interpolants.gamma) GaussianMixture (class in nami_toys.gaussian) GaussianRing (class in nami_toys.ring) GaussianShell (class in nami_toys.shell) generate() (nami_toys.checkerboard.Checkerboard method) (nami_toys.gaussian.GaussianMixture method) (nami_toys.moons.TwoMoons method) (nami_toys.parameterised.ParameterisedGaussian method) (nami_toys.ring.GaussianRing method) (nami_toys.shell.GaussianShell method) (nami_toys.spirals.TwoSpirals method) get_activation() (in module nami.components.activation)

H

has_rsample (nami.distributions.normal.DiagonalNormal attribute) (nami.distributions.normal.StandardNormal attribute)

has_rsample() (in module nami.distributions.base) Heun (class in nami.solvers.heun) HutchinsonDivergence (class in nami.divergence.hutchinson)

I

integrate() (nami.core.typing.ODESolver method) (nami.core.typing.SDESolver method) (nami.solvers.heun.Heun method) (nami.solvers.ode.RK4 method) (nami.solvers.sde.EulerMaruyama method) integrate_augmented() (nami.core.typing.ODESolver method) (nami.solvers.heun.Heun method) (nami.solvers.ode.RK4 method)

inverse() (nami_toys.standardise.Standardiser method) is_sde (nami.core.typing.ODESolver attribute) (nami.core.typing.SDESolver attribute) (nami.solvers.heun.Heun attribute) (nami.solvers.ode.RK4 attribute) (nami.solvers.sde.EulerMaruyama attribute)

L

LazyDistribution (class in nami.lazy) LazyField (class in nami.lazy) limit() (nami_toys.dataset.ToyDataset method) LinearPath (class in nami.paths.linear)

log_likelihood_ratio() (nami_toys.parameterised.ParameterisedGaussian method) log_prob() (nami.distributions.normal.DiagonalNormal method) (nami.distributions.normal.StandardNormal method) (nami.processes.fm.FlowMatchingProcess method) (nami_toys.parameterised.ParameterisedGaussian method)

M

make_generator() (in module nami_toys.rng) MarkovizationDriftFromVelocityScore (class in nami.interpolants.transforms) masked_fm_loss() (in module nami.masking) masked_sample() (in module nami.masking) mean (nami.distributions.normal.DiagonalNormal property) (nami.distributions.normal.StandardNormal property) (nami_toys.standardise.Standardiser attribute) meta (nami_toys.dataset.ToyDataset attribute) MirrorVelocityFromScore (class in nami.interpolants.transforms) MLPBackbone (class in nami.components.mlp) module nami.components.activation nami.components.mlp nami.components.time nami.components.transformer nami.core.broadcast nami.core.specs nami.core.typing nami.debug nami.diagnostics nami.distributions.base nami.distributions.normal nami.divergence.base nami.divergence.exact nami.divergence.hutchinson nami.fields.base nami.fields.diffusion nami.fields.transformer nami.fields.velocity nami.interpolants.gamma nami.interpolants.transforms nami.lazy nami.losses.fm nami.losses.stochastic_fm nami.masking nami.paths.base nami.paths.cosine nami.paths.linear nami.processes.diffusion nami.processes.fm nami.schedules.base nami.schedules.edm nami.schedules.ve nami.schedules.vp nami.solvers.heun nami.solvers.ode nami.solvers.sde nami_toys.checkerboard nami_toys.dataset nami_toys.gaussian nami_toys.moons nami_toys.parameterised nami_toys.ring nami_toys.rng nami_toys.shell nami_toys.spirals nami_toys.standardise

N

n_modes (nami_toys.ring.GaussianRing attribute) n_turns (nami_toys.spirals.TwoSpirals attribute) nami.components.activation module nami.components.mlp module nami.components.time module nami.components.transformer module nami.core.broadcast module nami.core.specs module nami.core.typing module nami.debug module nami.diagnostics module nami.distributions.base module nami.distributions.normal module nami.divergence.base module nami.divergence.exact module nami.divergence.hutchinson module nami.fields.base module nami.fields.diffusion module nami.fields.transformer module nami.fields.velocity module nami.interpolants.gamma module nami.interpolants.transforms module nami.lazy module nami.losses.fm module nami.losses.stochastic_fm module nami.masking module

nami.paths.base module nami.paths.cosine module nami.paths.linear module nami.processes.diffusion module nami.processes.fm module nami.schedules.base module nami.schedules.edm module nami.schedules.ve module nami.schedules.vp module nami.solvers.heun module nami.solvers.ode module nami.solvers.sde module nami_toys.checkerboard module nami_toys.dataset module nami_toys.gaussian module nami_toys.moons module nami_toys.parameterised module nami_toys.ring module nami_toys.rng module nami_toys.shell module nami_toys.spirals module nami_toys.standardise module noise (nami_toys.moons.TwoMoons attribute) (nami_toys.spirals.TwoSpirals attribute) NoiseSchedule (class in nami.core.typing) (class in nami.schedules.base) numel (nami.core.specs.TensorSpec property)

O

ODESolver (class in nami.core.typing)

out_dim (nami.components.time.ScalarTimeEmbedding attribute) (nami.components.time.SinusoidalTimeEmbedding property)

P

param_dim (nami_toys.parameterised.ParameterisedGaussian attribute) ParameterisedGaussian (class in nami_toys.parameterised)

ProbabilityPath (class in nami.core.typing) (class in nami.paths.base)

R

radius (nami_toys.ring.GaussianRing attribute) (nami_toys.shell.GaussianShell attribute) requires_steps (nami.core.typing.ODESolver attribute) (nami.solvers.heun.Heun attribute) (nami.solvers.ode.RK4 attribute) (nami.solvers.sde.EulerMaruyama attribute)

reversibility_error() (in module nami.diagnostics) RK4 (class in nami.solvers.ode) rsample() (nami.distributions.normal.DiagonalNormal method) (nami.distributions.normal.StandardNormal method) (nami.processes.diffusion.DiffusionProcess method) (nami.processes.fm.FlowMatchingProcess method)

S

sample() (nami.distributions.normal.DiagonalNormal method) (nami.distributions.normal.StandardNormal method) (nami.processes.diffusion.DiffusionProcess method) (nami.processes.fm.FlowMatchingProcess method) sample_xt() (nami.core.typing.ProbabilityPath method) (nami.paths.base.ProbabilityPath method) (nami.paths.cosine.CosinePath method) (nami.paths.linear.LinearPath method) ScalarTimeEmbedding (class in nami.components.time) scale (nami.interpolants.gamma.ScaledBrownianGamma attribute) ScaledBrownianGamma (class in nami.interpolants.gamma) score_model (nami.interpolants.transforms.DriftFromVelocityScore attribute) score_target() (nami.paths.base.ProbabilityPath method) score_to_eps() (in module nami.fields.diffusion) score_to_x0() (in module nami.fields.diffusion) ScoreFromNoise (class in nami.interpolants.transforms) SDESolver (class in nami.core.typing) sig (nami_toys.gaussian.GaussianMixture property) sig_at() (nami_toys.parameterised.ParameterisedGaussian method) sig_cov (nami_toys.gaussian.GaussianMixture attribute) (nami_toys.parameterised.ParameterisedGaussian attribute)

sig_frac (nami_toys.parameterised.ParameterisedGaussian attribute) sig_loc (nami_toys.gaussian.GaussianMixture attribute) (nami_toys.parameterised.ParameterisedGaussian attribute) sigma() (nami.core.typing.NoiseSchedule method) (nami.schedules.base.NoiseSchedule method) (nami.schedules.edm.EDMSchedule method) (nami.schedules.ve.VESchedule method) (nami.schedules.vp.VPSchedule method) SinusoidalTimeEmbedding (class in nami.components.time) snr() (nami.core.typing.NoiseSchedule method) (nami.schedules.base.NoiseSchedule method) split_event() (in module nami.core.specs) Standardiser (class in nami_toys.standardise) StandardNormal (class in nami.distributions.normal) std (nami_toys.ring.GaussianRing attribute) (nami_toys.standardise.Standardiser attribute) stochastic_fm_loss() (in module nami.losses.stochastic_fm) subset() (nami_toys.dataset.ToyDataset method) supports_rsample (nami.core.typing.ODESolver attribute) (nami.solvers.heun.Heun attribute) (nami.solvers.ode.RK4 attribute)

T

target_ut() (nami.core.typing.ProbabilityPath method) (nami.paths.base.ProbabilityPath method) (nami.paths.cosine.CosinePath method) (nami.paths.linear.LinearPath method) TensorSpec (class in nami.core.specs) ToyDataset (class in nami_toys.dataset)

transform() (nami_toys.standardise.Standardiser method) transform_dataset() (nami_toys.standardise.Standardiser method) TransformerBackbone (class in nami.components.transformer) TransformerBlock (class in nami.components.transformer) TransformerVelocityField (class in nami.fields.transformer) TwoMoons (class in nami_toys.moons) TwoSpirals (class in nami_toys.spirals)

U

UnconditionalDistribution (class in nami.lazy)

UnconditionalField (class in nami.lazy) unflatten_event() (in module nami.core.specs)

V

validate_shapes() (in module nami.core.specs) variance (nami.distributions.normal.DiagonalNormal property) (nami.distributions.normal.StandardNormal property) VectorField (class in nami.fields.base)

velocity_model (nami.interpolants.transforms.DriftFromVelocityScore attribute) VelocityField (class in nami.fields.velocity) VESchedule (class in nami.schedules.ve) VPSchedule (class in nami.schedules.vp)

W

width (nami_toys.shell.GaussianShell attribute)

X

x (nami_toys.dataset.ToyDataset attribute)

x0_to_eps() (in module nami.fields.diffusion) x0_to_score() (in module nami.fields.diffusion)

Y

y (nami_toys.dataset.ToyDataset attribute)

Z

ZeroGamma (class in nami.interpolants.gamma)