Index

A

AffineProbPath (class in gensbi.flow_matching.path) (class in gensbi.flow_matching.path.affine) alpha_t (gensbi.flow_matching.path.scheduler.scheduler.SchedulerOutput attribute), [1] (gensbi.flow_matching.path.scheduler.SchedulerOutput attribute), [1] apply_rope() (in module gensbi.models.flux1.math) assert_sample_shape() (gensbi.flow_matching.path.path.ProbPath method) (gensbi.flow_matching.path.ProbPath method)

attention() (in module gensbi.models.flux1.math) AttentionBlock (class in gensbi.models.simformer.transformer) axes_dim (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute)

B

BaseSDE (class in gensbi.diffusion.path.scheduler.edm)

BaseSDESolver (class in gensbi.flow_matching.solver.sde_solver)

C

c_in() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.EDMScheduler method) (gensbi.diffusion.path.scheduler.edm.VEScheduler method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.EDMScheduler method) (gensbi.diffusion.path.scheduler.VEScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) c_noise() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.EDMScheduler method) (gensbi.diffusion.path.scheduler.edm.VEScheduler method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.EDMScheduler method) (gensbi.diffusion.path.scheduler.VEScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) c_out() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.EDMScheduler method) (gensbi.diffusion.path.scheduler.edm.VEScheduler method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.EDMScheduler method) (gensbi.diffusion.path.scheduler.VEScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) c_skip() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.EDMScheduler method) (gensbi.diffusion.path.scheduler.edm.VEScheduler method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.EDMScheduler method) (gensbi.diffusion.path.scheduler.VEScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method)

cfg_scale (gensbi.utils.model_wrapping.GuidedModelWrapper attribute) cond_dim (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute) conditioned() (gensbi.models.simformer.simformer.SimformerConditioner method) (gensbi.models.simformer.SimformerConditioner method) (gensbi.models.SimformerConditioner method) CondOTProbPath (class in gensbi.flow_matching.path) (class in gensbi.flow_matching.path.affine) CondOTScheduler (class in gensbi.flow_matching.path.scheduler) (class in gensbi.flow_matching.path.scheduler.scheduler) context_in_dim (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute) ContinuousFMLoss (class in gensbi.flow_matching.loss) (class in gensbi.flow_matching.loss.continuous_loss) ConvexScheduler (class in gensbi.flow_matching.path.scheduler) (class in gensbi.flow_matching.path.scheduler.scheduler) CosineScheduler (class in gensbi.flow_matching.path.scheduler) (class in gensbi.flow_matching.path.scheduler.scheduler)

D

d_alpha_t (gensbi.flow_matching.path.scheduler.scheduler.SchedulerOutput attribute), [1] (gensbi.flow_matching.path.scheduler.SchedulerOutput attribute), [1] d_sigma_t (gensbi.flow_matching.path.scheduler.scheduler.SchedulerOutput attribute), [1] (gensbi.flow_matching.path.scheduler.SchedulerOutput attribute), [1] denoise() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) DenseBlock (class in gensbi.models.simformer.transformer) depth (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute) depth_single_blocks (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute) dim_condition (gensbi.models.simformer.simformer.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute) dim_id (gensbi.models.simformer.simformer.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute)

dim_joint (gensbi.models.simformer.simformer.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute) dim_value (gensbi.models.simformer.simformer.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute) DiscretePathSample (class in gensbi.flow_matching.path.path_sample) divergence() (in module gensbi.utils.math) DoubleStreamBlock (class in gensbi.models.flux1.layers) dropout_rate (gensbi.models.simformer.simformer.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute) dx_t (gensbi.flow_matching.path.path_sample.PathSample attribute), [1] (gensbi.flow_matching.path.PathSample attribute), [1]

E

edm_ablation_sampler() (in module gensbi.diffusion.solver.edm_samplers) edm_sampler() (in module gensbi.diffusion.solver.edm_samplers) EDMPath (class in gensbi.diffusion.path) (class in gensbi.diffusion.path.edm_path) EDMPathSample (class in gensbi.diffusion.path.path_sample) EDMScheduler (class in gensbi.diffusion.path.scheduler) (class in gensbi.diffusion.path.scheduler.edm) EmbedND (class in gensbi.models.flux1.layers)

epsilon_to_target() (gensbi.flow_matching.path.affine.AffineProbPath method) (gensbi.flow_matching.path.AffineProbPath method) epsilon_to_velocity() (gensbi.flow_matching.path.affine.AffineProbPath method) (gensbi.flow_matching.path.AffineProbPath method) Euclidean (class in gensbi.flow_matching.utils.manifolds_not_implemented.manifold) expand_tensor_like() (in module gensbi.flow_matching.utils) (in module gensbi.flow_matching.utils.utils) expmap() (gensbi.flow_matching.utils.manifolds_not_implemented.manifold.Euclidean method) (gensbi.flow_matching.utils.manifolds_not_implemented.manifold.Manifold method)

F

f() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) Flux (class in gensbi.models) (class in gensbi.models.flux1) (class in gensbi.models.flux1.model) FluxCFMLoss (class in gensbi.models) (class in gensbi.models.flux1) (class in gensbi.models.flux1.loss)

FluxParams (class in gensbi.models) (class in gensbi.models.flux1) (class in gensbi.models.flux1.model) fourier_features (gensbi.models.simformer.simformer.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute)

G

g() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) gate (gensbi.models.flux1.layers.ModulationOut attribute) GaussianFourierEmbedding (class in gensbi.models.simformer.embedding) gensbi module gensbi.diffusion module gensbi.diffusion.path module gensbi.diffusion.path.edm_path module gensbi.diffusion.path.path module gensbi.diffusion.path.path_sample module gensbi.diffusion.path.scheduler module gensbi.diffusion.path.scheduler.edm module gensbi.diffusion.solver module gensbi.diffusion.solver.edm_samplers module gensbi.diffusion.solver.sde_solver module gensbi.diffusion.solver.solver module gensbi.flow_matching module gensbi.flow_matching.loss module gensbi.flow_matching.loss.continuous_loss module gensbi.flow_matching.path module gensbi.flow_matching.path.affine module gensbi.flow_matching.path.path module gensbi.flow_matching.path.path_sample module gensbi.flow_matching.path.scheduler module gensbi.flow_matching.path.scheduler.schedule_transform module gensbi.flow_matching.path.scheduler.scheduler module gensbi.flow_matching.solver module gensbi.flow_matching.solver.ode_solver module gensbi.flow_matching.solver.sde_solver module gensbi.flow_matching.solver.solver module gensbi.flow_matching.solver.utils module gensbi.flow_matching.utils module gensbi.flow_matching.utils.manifolds_not_implemented.manifold module

gensbi.flow_matching.utils.utils module gensbi.models module gensbi.models.flux1 module gensbi.models.flux1.layers module gensbi.models.flux1.loss module gensbi.models.flux1.math module gensbi.models.flux1.model module gensbi.models.simformer module gensbi.models.simformer.embedding module gensbi.models.simformer.loss module gensbi.models.simformer.simformer module gensbi.models.simformer.transformer module gensbi.utils module gensbi.utils.math module gensbi.utils.model_wrapping module gensbi.utils.plotting module get_batch() (gensbi.diffusion.path.path_sample.EDMPathSample method) get_divergence() (gensbi.utils.model_wrapping.ModelWrapper method) get_f_tilde() (gensbi.flow_matching.solver.NonSingular method) (gensbi.flow_matching.solver.sde_solver.BaseSDESolver method) (gensbi.flow_matching.solver.sde_solver.NonSingular method) (gensbi.flow_matching.solver.sde_solver.ZeroEnds method) (gensbi.flow_matching.solver.ZeroEnds method) get_g_tilde() (gensbi.flow_matching.solver.NonSingular method) (gensbi.flow_matching.solver.sde_solver.BaseSDESolver method) (gensbi.flow_matching.solver.sde_solver.NonSingular method) (gensbi.flow_matching.solver.sde_solver.ZeroEnds method) (gensbi.flow_matching.solver.ZeroEnds method) get_loss_fn() (gensbi.diffusion.path.edm_path.EDMPath method) (gensbi.diffusion.path.EDMPath method) (gensbi.diffusion.path.scheduler.edm.BaseSDE method) get_nearest_times() (in module gensbi.flow_matching.solver.utils) get_sampler() (gensbi.diffusion.solver.sde_solver.SDESolver method) (gensbi.diffusion.solver.SDESolver method) (gensbi.flow_matching.solver.ode_solver.ODESolver method) (gensbi.flow_matching.solver.ODESolver method) (gensbi.flow_matching.solver.sde_solver.BaseSDESolver method) get_score() (gensbi.flow_matching.solver.sde_solver.BaseSDESolver method) get_score_function() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) get_unnormalized_logprob() (gensbi.flow_matching.solver.ode_solver.ODESolver method) (gensbi.flow_matching.solver.ODESolver method) get_vector_field() (gensbi.utils.model_wrapping.GuidedModelWrapper method) (gensbi.utils.model_wrapping.ModelWrapper method) guidance_embed (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute) GuidedModelWrapper (class in gensbi.utils.model_wrapping)

I

Identity (class in gensbi.models.flux1.model) in_channels (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute)

K

kappa_inverse() (gensbi.flow_matching.path.scheduler.CondOTScheduler method) (gensbi.flow_matching.path.scheduler.ConvexScheduler method) (gensbi.flow_matching.path.scheduler.PolynomialConvexScheduler method) (gensbi.flow_matching.path.scheduler.scheduler.CondOTScheduler method) (gensbi.flow_matching.path.scheduler.scheduler.ConvexScheduler method) (gensbi.flow_matching.path.scheduler.scheduler.PolynomialConvexScheduler method)

L

LastLayer (class in gensbi.models.flux1.layers) LinearVPScheduler (class in gensbi.flow_matching.path.scheduler) (class in gensbi.flow_matching.path.scheduler.scheduler) logmap() (gensbi.flow_matching.utils.manifolds_not_implemented.manifold.Euclidean method) (gensbi.flow_matching.utils.manifolds_not_implemented.manifold.Manifold method) loss_weight() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.EDMScheduler method) (gensbi.diffusion.path.scheduler.edm.VEScheduler method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.EDMScheduler method) (gensbi.diffusion.path.scheduler.VEScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method)

M

Manifold (class in gensbi.flow_matching.utils.manifolds_not_implemented.manifold) mlp_ratio (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute) MLPEmbedder (class in gensbi.models.flux1.layers) (class in gensbi.models.simformer.embedding) ModelWrapper (class in gensbi.utils.model_wrapping) Modulation (class in gensbi.models.flux1.layers) ModulationOut (class in gensbi.models.flux1.layers) module gensbi gensbi.diffusion gensbi.diffusion.path gensbi.diffusion.path.edm_path gensbi.diffusion.path.path gensbi.diffusion.path.path_sample gensbi.diffusion.path.scheduler gensbi.diffusion.path.scheduler.edm gensbi.diffusion.solver gensbi.diffusion.solver.edm_samplers gensbi.diffusion.solver.sde_solver gensbi.diffusion.solver.solver gensbi.flow_matching gensbi.flow_matching.loss gensbi.flow_matching.loss.continuous_loss gensbi.flow_matching.path gensbi.flow_matching.path.affine gensbi.flow_matching.path.path gensbi.flow_matching.path.path_sample gensbi.flow_matching.path.scheduler gensbi.flow_matching.path.scheduler.schedule_transform gensbi.flow_matching.path.scheduler.scheduler gensbi.flow_matching.solver gensbi.flow_matching.solver.ode_solver gensbi.flow_matching.solver.sde_solver gensbi.flow_matching.solver.solver gensbi.flow_matching.solver.utils gensbi.flow_matching.utils gensbi.flow_matching.utils.manifolds_not_implemented.manifold gensbi.flow_matching.utils.utils gensbi.models gensbi.models.flux1 gensbi.models.flux1.layers gensbi.models.flux1.loss gensbi.models.flux1.math gensbi.models.flux1.model gensbi.models.simformer gensbi.models.simformer.embedding gensbi.models.simformer.loss gensbi.models.simformer.simformer gensbi.models.simformer.transformer gensbi.utils gensbi.utils.math gensbi.utils.model_wrapping gensbi.utils.plotting

N

name (gensbi.diffusion.path.path.ProbPath property) (gensbi.diffusion.path.scheduler.edm.BaseSDE property) (gensbi.diffusion.path.scheduler.edm.EDMScheduler property) (gensbi.diffusion.path.scheduler.edm.VEScheduler property) (gensbi.diffusion.path.scheduler.edm.VPScheduler property) (gensbi.diffusion.path.scheduler.EDMScheduler property) (gensbi.diffusion.path.scheduler.VEScheduler property) (gensbi.diffusion.path.scheduler.VPScheduler property) NonSingular (class in gensbi.flow_matching.solver) (class in gensbi.flow_matching.solver.sde_solver) num_heads (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute) (gensbi.models.simformer.simformer.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute)

num_hidden_layers (gensbi.models.simformer.simformer.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute) num_layers (gensbi.models.simformer.simformer.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute)

O

obs_dim (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute)

ODESolver (class in gensbi.flow_matching.solver) (class in gensbi.flow_matching.solver.ode_solver)

P

param_dtype (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute) PathSample (class in gensbi.flow_matching.path) (class in gensbi.flow_matching.path.path_sample) plot_trajectories() (in module gensbi.utils.plotting) PolynomialConvexScheduler (class in gensbi.flow_matching.path.scheduler) (class in gensbi.flow_matching.path.scheduler.scheduler)

ProbPath (class in gensbi.diffusion.path.path) (class in gensbi.flow_matching.path) (class in gensbi.flow_matching.path.path) proju() (gensbi.flow_matching.utils.manifolds_not_implemented.manifold.Euclidean method) (gensbi.flow_matching.utils.manifolds_not_implemented.manifold.Manifold method) projx() (gensbi.flow_matching.utils.manifolds_not_implemented.manifold.Euclidean method) (gensbi.flow_matching.utils.manifolds_not_implemented.manifold.Manifold method)

Q

QKNorm (class in gensbi.models.flux1.layers) qkv_bias (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute) qkv_features (gensbi.models.simformer.simformer.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute)

qkv_multiplier (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute)

R

rngs (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute) (gensbi.models.simformer.simformer.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute)

rope() (in module gensbi.models.flux1.math)

S

s() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.EDMScheduler method) (gensbi.diffusion.path.scheduler.edm.VEScheduler method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.EDMScheduler method) (gensbi.diffusion.path.scheduler.VEScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) s_deriv() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.EDMScheduler method) (gensbi.diffusion.path.scheduler.edm.VEScheduler method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.EDMScheduler method) (gensbi.diffusion.path.scheduler.VEScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) sample() (gensbi.diffusion.path.edm_path.EDMPath method) (gensbi.diffusion.path.EDMPath method) (gensbi.diffusion.path.path.ProbPath method) (gensbi.diffusion.solver.sde_solver.SDESolver method) (gensbi.diffusion.solver.SDESolver method) (gensbi.diffusion.solver.Solver method) (gensbi.diffusion.solver.solver.Solver method) (gensbi.flow_matching.path.affine.AffineProbPath method) (gensbi.flow_matching.path.AffineProbPath method) (gensbi.flow_matching.path.path.ProbPath method) (gensbi.flow_matching.path.ProbPath method) (gensbi.flow_matching.solver.ode_solver.ODESolver method) (gensbi.flow_matching.solver.ODESolver method) (gensbi.flow_matching.solver.sde_solver.BaseSDESolver method) (gensbi.flow_matching.solver.Solver method) (gensbi.flow_matching.solver.solver.Solver method) sample_noise() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) sample_prior() (gensbi.diffusion.path.path.ProbPath method) (gensbi.diffusion.path.scheduler.edm.BaseSDE method) sample_sigma() (gensbi.diffusion.path.edm_path.EDMPath method) (gensbi.diffusion.path.EDMPath method) (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.EDMScheduler method) (gensbi.diffusion.path.scheduler.edm.VEScheduler method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.EDMScheduler method) (gensbi.diffusion.path.scheduler.VEScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) scale (gensbi.models.flux1.layers.ModulationOut attribute) Scheduler (class in gensbi.flow_matching.path.scheduler) (class in gensbi.flow_matching.path.scheduler.scheduler) SchedulerOutput (class in gensbi.flow_matching.path.scheduler) (class in gensbi.flow_matching.path.scheduler.scheduler) ScheduleTransformedModel (class in gensbi.flow_matching.path.scheduler) (class in gensbi.flow_matching.path.scheduler.schedule_transform) SDESolver (class in gensbi.diffusion.solver) (class in gensbi.diffusion.solver.sde_solver) SelfAttention (class in gensbi.models.flux1.layers) shift (gensbi.models.flux1.layers.ModulationOut attribute)

sigma (gensbi.diffusion.path.path_sample.EDMPathSample attribute), [1] sigma() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.EDMScheduler method) (gensbi.diffusion.path.scheduler.edm.VEScheduler method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.EDMScheduler method) (gensbi.diffusion.path.scheduler.VEScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) sigma_deriv() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.EDMScheduler method) (gensbi.diffusion.path.scheduler.edm.VEScheduler method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.EDMScheduler method) (gensbi.diffusion.path.scheduler.VEScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) sigma_inv() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.EDMScheduler method) (gensbi.diffusion.path.scheduler.edm.VEScheduler method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.EDMScheduler method) (gensbi.diffusion.path.scheduler.VEScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) sigma_t (gensbi.flow_matching.path.scheduler.scheduler.SchedulerOutput attribute), [1] (gensbi.flow_matching.path.scheduler.SchedulerOutput attribute), [1] Simformer (class in gensbi.models) (class in gensbi.models.simformer) (class in gensbi.models.simformer.simformer) SimformerCFMLoss (class in gensbi.models) (class in gensbi.models.simformer) (class in gensbi.models.simformer.loss) SimformerConditioner (class in gensbi.models) (class in gensbi.models.simformer) (class in gensbi.models.simformer.simformer) SimformerParams (class in gensbi.models) (class in gensbi.models.simformer) (class in gensbi.models.simformer.simformer) SimpleTimeEmbedding (class in gensbi.models.simformer.embedding) SingleStreamBlock (class in gensbi.models.flux1.layers) SinusoidalEmbedding (class in gensbi.models.simformer.embedding) snr_inverse() (gensbi.flow_matching.path.scheduler.ConvexScheduler method) (gensbi.flow_matching.path.scheduler.CosineScheduler method) (gensbi.flow_matching.path.scheduler.LinearVPScheduler method) (gensbi.flow_matching.path.scheduler.Scheduler method) (gensbi.flow_matching.path.scheduler.scheduler.ConvexScheduler method) (gensbi.flow_matching.path.scheduler.scheduler.CosineScheduler method) (gensbi.flow_matching.path.scheduler.scheduler.LinearVPScheduler method) (gensbi.flow_matching.path.scheduler.scheduler.Scheduler method) (gensbi.flow_matching.path.scheduler.scheduler.VPScheduler method) (gensbi.flow_matching.path.scheduler.VPScheduler method) Solver (class in gensbi.diffusion.solver) (class in gensbi.diffusion.solver.solver) (class in gensbi.flow_matching.solver) (class in gensbi.flow_matching.solver.solver)

T

t (gensbi.flow_matching.path.path_sample.DiscretePathSample attribute), [1] (gensbi.flow_matching.path.path_sample.PathSample attribute), [1] (gensbi.flow_matching.path.PathSample attribute), [1] target_to_epsilon() (gensbi.flow_matching.path.affine.AffineProbPath method) (gensbi.flow_matching.path.AffineProbPath method) target_to_velocity() (gensbi.flow_matching.path.affine.AffineProbPath method) (gensbi.flow_matching.path.AffineProbPath method) theta (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute)

time_schedule() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.EDMScheduler method) (gensbi.diffusion.path.scheduler.edm.VEScheduler method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.EDMScheduler method) (gensbi.diffusion.path.scheduler.VEScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) timestep_embedding() (in module gensbi.models.flux1.layers) timesteps() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) Transformer (class in gensbi.models.simformer.transformer)

U

unconditioned() (gensbi.models.simformer.simformer.SimformerConditioner method) (gensbi.models.simformer.SimformerConditioner method) (gensbi.models.SimformerConditioner method) unnormalized_logprob() (gensbi.flow_matching.solver.ode_solver.ODESolver method) (gensbi.flow_matching.solver.ODESolver method)

unsqueeze_to_match() (in module gensbi.flow_matching.utils) (in module gensbi.flow_matching.utils.utils) use_rope (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute)

V

vec_in_dim (gensbi.models.flux1.FluxParams attribute) (gensbi.models.flux1.model.FluxParams attribute) (gensbi.models.FluxParams attribute) velocity_to_epsilon() (gensbi.flow_matching.path.affine.AffineProbPath method) (gensbi.flow_matching.path.AffineProbPath method) velocity_to_target() (gensbi.flow_matching.path.affine.AffineProbPath method) (gensbi.flow_matching.path.AffineProbPath method)

VEScheduler (class in gensbi.diffusion.path.scheduler) (class in gensbi.diffusion.path.scheduler.edm) VPScheduler (class in gensbi.diffusion.path.scheduler) (class in gensbi.diffusion.path.scheduler.edm) (class in gensbi.flow_matching.path.scheduler) (class in gensbi.flow_matching.path.scheduler.scheduler)

W

widening_factor (gensbi.models.simformer.simformer.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute)

X

x_0 (gensbi.flow_matching.path.path_sample.DiscretePathSample attribute), [1] (gensbi.flow_matching.path.path_sample.PathSample attribute), [1] (gensbi.flow_matching.path.PathSample attribute), [1] x_1 (gensbi.diffusion.path.path_sample.EDMPathSample attribute), [1] (gensbi.flow_matching.path.path_sample.DiscretePathSample attribute), [1] (gensbi.flow_matching.path.path_sample.PathSample attribute), [1] (gensbi.flow_matching.path.PathSample attribute), [1]

x_t (gensbi.diffusion.path.path_sample.EDMPathSample attribute), [1] (gensbi.flow_matching.path.path_sample.DiscretePathSample attribute), [1] (gensbi.flow_matching.path.path_sample.PathSample attribute), [1] (gensbi.flow_matching.path.PathSample attribute), [1]

Z

ZeroEnds (class in gensbi.flow_matching.solver) (class in gensbi.flow_matching.solver.sde_solver)