Index

_

__call__() (gensbi.experimental.models.AutoEncoder1D method) (gensbi.experimental.models.AutoEncoder2D method) (gensbi.experimental.models.autoencoders.AutoEncoder1D method) (gensbi.experimental.models.autoencoders.AutoEncoder2D method) (gensbi.experimental.models.autoencoders.autoencoder_1d.AttnBlock1D method) (gensbi.experimental.models.autoencoders.autoencoder_1d.AutoEncoder1D method) (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D method) (gensbi.experimental.models.autoencoders.autoencoder_1d.Downsample1D method) (gensbi.experimental.models.autoencoders.autoencoder_1d.Encoder1D method) (gensbi.experimental.models.autoencoders.autoencoder_1d.ResnetBlock1D method) (gensbi.experimental.models.autoencoders.autoencoder_1d.Upsample1D method) (gensbi.experimental.models.autoencoders.autoencoder_2d.AttnBlock2D method) (gensbi.experimental.models.autoencoders.autoencoder_2d.AutoEncoder2D method) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D method) (gensbi.experimental.models.autoencoders.autoencoder_2d.Downsample2D method) (gensbi.experimental.models.autoencoders.autoencoder_2d.Encoder2D method) (gensbi.experimental.models.autoencoders.autoencoder_2d.ResnetBlock2D method) (gensbi.experimental.models.autoencoders.autoencoder_2d.Upsample2D method) (gensbi.experimental.models.autoencoders.commons.DiagonalGaussian method) (gensbi.flow_matching.loss.continuous_loss.ContinuousFMLoss method) (gensbi.flow_matching.loss.ContinuousFMLoss method) (gensbi.flow_matching.path.scheduler.CondOTScheduler method) (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.PolynomialConvexScheduler method) (gensbi.flow_matching.path.scheduler.schedule_transform.ScheduleTransformedModel method) (gensbi.flow_matching.path.scheduler.Scheduler method) (gensbi.flow_matching.path.scheduler.scheduler.CondOTScheduler 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.PolynomialConvexScheduler method) (gensbi.flow_matching.path.scheduler.scheduler.Scheduler method) (gensbi.flow_matching.path.scheduler.scheduler.VPScheduler method) (gensbi.flow_matching.path.scheduler.ScheduleTransformedModel method) (gensbi.flow_matching.path.scheduler.VPScheduler method) (gensbi.models.ConditionalCFMLoss method) (gensbi.models.ConditionalDiffLoss method) (gensbi.models.ConditionalWrapper method) (gensbi.models.embedding.Embed method) (gensbi.models.embedding.embedding.Embed method) (gensbi.models.embedding.embedding.FeatureEmbedder method) (gensbi.models.embedding.embedding.GaussianFourierEmbedding method) (gensbi.models.embedding.embedding.MLPEmbedder method) (gensbi.models.embedding.embedding.SimpleTimeEmbedding method) (gensbi.models.embedding.embedding.SinusoidalPosEmbed1D method) (gensbi.models.embedding.embedding.SinusoidalPosEmbed2D method) (gensbi.models.embedding.embedding.SinusoidalTimeEmbedding method) (gensbi.models.embedding.FeatureEmbedder method) (gensbi.models.embedding.GaussianFourierEmbedding method) (gensbi.models.embedding.MLPEmbedder method) (gensbi.models.embedding.SimpleTimeEmbedding method) (gensbi.models.embedding.SinusoidalPosEmbed1D method) (gensbi.models.embedding.SinusoidalPosEmbed2D method) (gensbi.models.embedding.SinusoidalTimeEmbedding method) (gensbi.models.Flux1 method) (gensbi.models.flux1.Flux1 method) (gensbi.models.flux1.layers.DoubleStreamBlock method) (gensbi.models.flux1.layers.EmbedND method) (gensbi.models.flux1.layers.Identity method) (gensbi.models.flux1.layers.LastLayer method) (gensbi.models.flux1.layers.MLPEmbedder method) (gensbi.models.flux1.layers.Modulation method) (gensbi.models.flux1.layers.QKNorm method) (gensbi.models.flux1.layers.SelfAttention method) (gensbi.models.flux1.layers.SingleStreamBlock method) (gensbi.models.flux1.model.Flux1 method) (gensbi.models.Flux1Joint method) (gensbi.models.flux1joint.Flux1Joint method) (gensbi.models.flux1joint.model.Flux1Joint method) (gensbi.models.JointCFMLoss method) (gensbi.models.JointDiffLoss method) (gensbi.models.JointWrapper method) (gensbi.models.losses.conditional.ConditionalCFMLoss method) (gensbi.models.losses.conditional.ConditionalDiffLoss method) (gensbi.models.losses.ConditionalCFMLoss method) (gensbi.models.losses.ConditionalDiffLoss method) (gensbi.models.losses.joint.JointCFMLoss method) (gensbi.models.losses.joint.JointDiffLoss method) (gensbi.models.losses.JointCFMLoss method) (gensbi.models.losses.JointDiffLoss method) (gensbi.models.losses.unconditional.UnconditionalCFMLoss method) (gensbi.models.losses.unconditional.UnconditionalDiffLoss method) (gensbi.models.losses.UnconditionalCFMLoss method) (gensbi.models.losses.UnconditionalDiffLoss method) (gensbi.models.Simformer method) (gensbi.models.simformer.model.Simformer method) (gensbi.models.simformer.Simformer method) (gensbi.models.simformer.transformer.AttentionBlock method) (gensbi.models.simformer.transformer.DenseBlock method) (gensbi.models.simformer.transformer.Transformer method) (gensbi.models.UnconditionalCFMLoss method) (gensbi.models.UnconditionalDiffLoss method) (gensbi.models.UnconditionalWrapper method) (gensbi.models.wrappers.conditional.ConditionalWrapper method) (gensbi.models.wrappers.ConditionalWrapper method) (gensbi.models.wrappers.joint.JointWrapper method) (gensbi.models.wrappers.JointWrapper method) (gensbi.models.wrappers.unconditional.UnconditionalWrapper method) (gensbi.models.wrappers.UnconditionalWrapper method) (gensbi.utils.model_wrapping.ModelWrapper method)

__post_init__() (gensbi.models.flux1.Flux1Params method) (gensbi.models.flux1.model.Flux1Params method) (gensbi.models.flux1joint.Flux1JointParams method) (gensbi.models.flux1joint.model.Flux1JointParams method) (gensbi.models.Flux1JointParams method) (gensbi.models.Flux1Params method) (gensbi.models.simformer.model.SimformerParams method) (gensbi.models.simformer.SimformerParams method) (gensbi.models.SimformerParams method) __version__ (in module gensbi) _expand_dims() (in module gensbi.utils.math) _expand_time() (in module gensbi.utils.math) _get_batch_sampler() (in module gensbi.recipes.pipeline) _get_default_params() (gensbi.recipes.conditional_pipeline.ConditionalDiffusionPipeline method) (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline method) (gensbi.recipes.ConditionalDiffusionPipeline method) (gensbi.recipes.ConditionalFlowPipeline method) (gensbi.recipes.flux1.Flux1DiffusionPipeline method) (gensbi.recipes.flux1.Flux1FlowPipeline method) (gensbi.recipes.Flux1DiffusionPipeline method) (gensbi.recipes.Flux1FlowPipeline method) (gensbi.recipes.flux1joint.Flux1JointDiffusionPipeline method) (gensbi.recipes.flux1joint.Flux1JointFlowPipeline method) (gensbi.recipes.Flux1JointDiffusionPipeline method) (gensbi.recipes.Flux1JointFlowPipeline method) (gensbi.recipes.joint_pipeline.JointDiffusionPipeline method) (gensbi.recipes.joint_pipeline.JointFlowPipeline method) (gensbi.recipes.JointDiffusionPipeline method) (gensbi.recipes.JointFlowPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) (gensbi.recipes.simformer.SimformerDiffusionPipeline method) (gensbi.recipes.simformer.SimformerFlowPipeline method) (gensbi.recipes.SimformerDiffusionPipeline method) (gensbi.recipes.SimformerFlowPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalDiffusionPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalFlowPipeline method) (gensbi.recipes.UnconditionalDiffusionPipeline method) (gensbi.recipes.UnconditionalFlowPipeline method) _get_ema_optimizer() (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) _get_kl_schedule() (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline method) _get_optimizer() (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) _make_model() (gensbi.recipes.conditional_pipeline.ConditionalDiffusionPipeline method) (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline method) (gensbi.recipes.ConditionalDiffusionPipeline method) (gensbi.recipes.ConditionalFlowPipeline method) (gensbi.recipes.flux1.Flux1DiffusionPipeline method) (gensbi.recipes.flux1.Flux1FlowPipeline method) (gensbi.recipes.Flux1DiffusionPipeline method) (gensbi.recipes.Flux1FlowPipeline method) (gensbi.recipes.flux1joint.Flux1JointDiffusionPipeline method) (gensbi.recipes.flux1joint.Flux1JointFlowPipeline method) (gensbi.recipes.Flux1JointDiffusionPipeline method) (gensbi.recipes.Flux1JointFlowPipeline method) (gensbi.recipes.joint_pipeline.JointDiffusionPipeline method) (gensbi.recipes.joint_pipeline.JointFlowPipeline method) (gensbi.recipes.JointDiffusionPipeline method) (gensbi.recipes.JointFlowPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) (gensbi.recipes.simformer.SimformerDiffusionPipeline method) (gensbi.recipes.simformer.SimformerFlowPipeline method) (gensbi.recipes.SimformerDiffusionPipeline method) (gensbi.recipes.SimformerFlowPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalDiffusionPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalFlowPipeline method) (gensbi.recipes.UnconditionalDiffusionPipeline method) (gensbi.recipes.UnconditionalFlowPipeline method) _parse_range() (in module gensbi.utils.plotting) _plot_cdf_region_expected_under_uniformity() (in module gensbi.diagnostics.sbc) _plot_hist_region_expected_under_uniformity() (in module gensbi.diagnostics.sbc) _plot_marginals_corner() (in module gensbi.utils.plotting) _plot_marginals_seaborn() (in module gensbi.utils.plotting) _plot_ranks_as_cdf() (in module gensbi.diagnostics.sbc) _plot_ranks_as_hist() (in module gensbi.diagnostics.sbc) _prepare_reduce_functions() (in module gensbi.diagnostics.sbc) _process_x() (gensbi.diagnostics.distribution_wrapper.PosteriorWrapper method) (gensbi.diagnostics.PosteriorWrapper method) _ravel() (gensbi.diagnostics.distribution_wrapper.PosteriorWrapper method) (gensbi.diagnostics.PosteriorWrapper method) _run_sbc() (in module gensbi.diagnostics.sbc) _run_tarp() (in module gensbi.diagnostics.tarp) _sbc_rank_plot() (in module gensbi.diagnostics.sbc) _train() (gensbi.diagnostics.LC2ST method) (gensbi.diagnostics.lc2st.LC2ST method) _unravel_theta() (gensbi.diagnostics.distribution_wrapper.PosteriorWrapper method) (gensbi.diagnostics.PosteriorWrapper method) _unravel_xs() (gensbi.diagnostics.distribution_wrapper.PosteriorWrapper method) (gensbi.diagnostics.PosteriorWrapper method) _validate_sbc_inputs() (in module gensbi.diagnostics.sbc) _wrap_model() (gensbi.recipes.conditional_pipeline.ConditionalDiffusionPipeline method) (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline method) (gensbi.recipes.ConditionalDiffusionPipeline method) (gensbi.recipes.ConditionalFlowPipeline method) (gensbi.recipes.joint_pipeline.JointDiffusionPipeline method) (gensbi.recipes.joint_pipeline.JointFlowPipeline method) (gensbi.recipes.JointDiffusionPipeline method) (gensbi.recipes.JointFlowPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalDiffusionPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalFlowPipeline method) (gensbi.recipes.UnconditionalDiffusionPipeline method) (gensbi.recipes.UnconditionalFlowPipeline method)

A

AbstractPipeline (class in gensbi.recipes.pipeline) AbstractVAEPipeline (class in gensbi.experimental.recipes.vae_pipeline) act (gensbi.models.simformer.transformer.DenseBlock attribute) (gensbi.models.simformer.transformer.Transformer attribute) adaLN_modulation (gensbi.models.flux1.layers.LastLayer attribute) 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) args (gensbi.diagnostics.distribution_wrapper.PosteriorWrapper attribute) (gensbi.diagnostics.PosteriorWrapper attribute) assert_sample_shape() (gensbi.flow_matching.path.path.ProbPath method) (gensbi.flow_matching.path.ProbPath method) attention() (gensbi.experimental.models.autoencoders.autoencoder_1d.AttnBlock1D method) (gensbi.experimental.models.autoencoders.autoencoder_2d.AttnBlock2D method) (in module gensbi.models.flux1.math) attention_blocks (gensbi.models.simformer.transformer.Transformer attribute) AttentionBlock (class in gensbi.models.simformer.transformer)

attn (gensbi.models.simformer.transformer.AttentionBlock attribute) AttnBlock1D (class in gensbi.experimental.models.autoencoders.autoencoder_1d) AttnBlock2D (class in gensbi.experimental.models.autoencoders.autoencoder_2d) AutoEncoder1D (class in gensbi.experimental.models) (class in gensbi.experimental.models.autoencoders) (class in gensbi.experimental.models.autoencoders.autoencoder_1d) AutoEncoder2D (class in gensbi.experimental.models) (class in gensbi.experimental.models.autoencoders) (class in gensbi.experimental.models.autoencoders.autoencoder_2d) AutoEncoderParams (class in gensbi.experimental.models) (class in gensbi.experimental.models.autoencoders) (class in gensbi.experimental.models.autoencoders.commons) axes_dim (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.layers.EmbedND attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.Flux1Params attribute)

B

B (gensbi.models.embedding.embedding.GaussianFourierEmbedding attribute) (gensbi.models.embedding.GaussianFourierEmbedding attribute) base_color (in module gensbi.utils.plotting) BaseSDE (class in gensbi.diffusion.path.scheduler.edm) beta_d (gensbi.diffusion.path.scheduler.edm.VPScheduler attribute) (gensbi.diffusion.path.scheduler.VPScheduler attribute) beta_max (gensbi.diffusion.path.scheduler.edm.VPScheduler attribute) (gensbi.diffusion.path.scheduler.VPScheduler attribute) (gensbi.flow_matching.path.scheduler.scheduler.VPScheduler attribute) (gensbi.flow_matching.path.scheduler.VPScheduler attribute)

beta_min (gensbi.diffusion.path.scheduler.edm.VPScheduler attribute) (gensbi.diffusion.path.scheduler.VPScheduler attribute) (gensbi.flow_matching.path.scheduler.scheduler.VPScheduler attribute) (gensbi.flow_matching.path.scheduler.VPScheduler attribute)

C

c2st() (in module gensbi.diagnostics.metrics) (in module gensbi.diagnostics.metrics.c2st) 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.models.ConditionalCFMLoss attribute) (gensbi.models.losses.conditional.ConditionalCFMLoss attribute) (gensbi.models.losses.ConditionalCFMLoss attribute) ch (gensbi.experimental.models.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.Encoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Encoder2D attribute) (gensbi.experimental.models.autoencoders.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.commons.AutoEncoderParams attribute), [1] ch_cond (gensbi.recipes.flux1.Flux1DiffusionPipeline attribute) (gensbi.recipes.flux1.Flux1FlowPipeline attribute) (gensbi.recipes.Flux1DiffusionPipeline attribute) (gensbi.recipes.Flux1FlowPipeline attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) ch_mult (gensbi.experimental.models.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.commons.AutoEncoderParams attribute), [1] ch_obs (gensbi.recipes.flux1.Flux1DiffusionPipeline attribute) (gensbi.recipes.flux1.Flux1FlowPipeline attribute) (gensbi.recipes.Flux1DiffusionPipeline attribute) (gensbi.recipes.Flux1FlowPipeline attribute) (gensbi.recipes.flux1joint.Flux1JointDiffusionPipeline attribute) (gensbi.recipes.flux1joint.Flux1JointFlowPipeline attribute) (gensbi.recipes.Flux1JointDiffusionPipeline attribute) (gensbi.recipes.Flux1JointFlowPipeline attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) (gensbi.recipes.simformer.SimformerDiffusionPipeline attribute) (gensbi.recipes.simformer.SimformerFlowPipeline attribute) (gensbi.recipes.SimformerDiffusionPipeline attribute) (gensbi.recipes.SimformerFlowPipeline attribute) check_c2st() (in module gensbi.diagnostics.metrics) (in module gensbi.diagnostics.metrics.c2st) check_prior_vs_dap() (in module gensbi.diagnostics.sbc) check_sbc() (in module gensbi.diagnostics) (in module gensbi.diagnostics.sbc) check_tarp() (in module gensbi.diagnostics) (in module gensbi.diagnostics.tarp) check_uniformity_c2st() (in module gensbi.diagnostics.sbc) check_uniformity_frequentist() (in module gensbi.diagnostics.sbc) chunk_dim (gensbi.experimental.models.autoencoders.commons.DiagonalGaussian attribute) clf (gensbi.diagnostics.lc2st.EnsembleClassifier attribute) clf_class (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute)

cmap_lcontour (in module gensbi.utils.plotting) colors (in module gensbi.utils.plotting) cond_attn (gensbi.models.flux1.layers.DoubleStreamBlock attribute) cond_ids (gensbi.recipes.conditional_pipeline.ConditionalDiffusionPipeline attribute) (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline attribute) (gensbi.recipes.ConditionalDiffusionPipeline attribute) (gensbi.recipes.ConditionalFlowPipeline attribute) cond_in (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.model.Flux1 attribute) cond_mlp (gensbi.models.flux1.layers.DoubleStreamBlock attribute) cond_mod (gensbi.models.flux1.layers.DoubleStreamBlock attribute) cond_norm1 (gensbi.models.flux1.layers.DoubleStreamBlock attribute) cond_norm2 (gensbi.models.flux1.layers.DoubleStreamBlock attribute) condition_dim (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) condition_embedding (gensbi.models.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1Joint attribute) (gensbi.models.flux1joint.model.Flux1Joint attribute) (gensbi.models.Simformer attribute) (gensbi.models.simformer.model.Simformer attribute) (gensbi.models.simformer.Simformer attribute) condition_mask_kind (gensbi.recipes.joint_pipeline.JointDiffusionPipeline attribute) (gensbi.recipes.joint_pipeline.JointFlowPipeline attribute) (gensbi.recipes.JointDiffusionPipeline attribute) (gensbi.recipes.JointFlowPipeline attribute) ConditionalCFMLoss (class in gensbi.models) (class in gensbi.models.losses) (class in gensbi.models.losses.conditional) ConditionalDiffLoss (class in gensbi.models) (class in gensbi.models.losses) (class in gensbi.models.losses.conditional) ConditionalDiffusionPipeline (class in gensbi.recipes) (class in gensbi.recipes.conditional_pipeline) ConditionalFlowPipeline (class in gensbi.recipes) (class in gensbi.recipes.conditional_pipeline) ConditionalWrapper (class in gensbi.models) (class in gensbi.models.wrappers) (class in gensbi.models.wrappers.conditional) conditioned() (gensbi.models.JointWrapper method) (gensbi.models.wrappers.joint.JointWrapper method) (gensbi.models.wrappers.JointWrapper 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_block (gensbi.models.simformer.transformer.DenseBlock attribute) context_in_dim (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.Flux1Params attribute) ContinuousFMLoss (class in gensbi.flow_matching.loss) (class in gensbi.flow_matching.loss.continuous_loss) conv (gensbi.experimental.models.autoencoders.autoencoder_1d.Downsample1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.Upsample1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Downsample2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Upsample2D attribute) conv1 (gensbi.experimental.models.autoencoders.autoencoder_1d.ResnetBlock1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.ResnetBlock2D attribute) conv2 (gensbi.experimental.models.autoencoders.autoencoder_1d.ResnetBlock1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.ResnetBlock2D attribute) conv_in (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.Encoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Encoder2D attribute) conv_out (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.Encoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Encoder2D attribute) 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] dcontext (gensbi.models.simformer.transformer.Transformer attribute) decode() (gensbi.experimental.models.AutoEncoder1D method) (gensbi.experimental.models.AutoEncoder2D method) (gensbi.experimental.models.autoencoders.AutoEncoder1D method) (gensbi.experimental.models.autoencoders.AutoEncoder2D method) (gensbi.experimental.models.autoencoders.autoencoder_1d.AutoEncoder1D method) (gensbi.experimental.models.autoencoders.autoencoder_2d.AutoEncoder2D method) Decoder1D (class in gensbi.experimental.models.autoencoders.autoencoder_1d) (gensbi.experimental.models.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.AutoEncoder1D attribute) Decoder2D (class in gensbi.experimental.models.autoencoders.autoencoder_2d) (gensbi.experimental.models.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AutoEncoder2D attribute) default_x (gensbi.diagnostics.distribution_wrapper.PosteriorWrapper attribute) (gensbi.diagnostics.PosteriorWrapper attribute) denoise() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) dense_blocks (gensbi.models.simformer.transformer.Transformer attribute) DenseBlock (class in gensbi.models.simformer.transformer) depth (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.Flux1Params attribute) depth_single_blocks (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.Flux1Params attribute) DiagonalGaussian (class in gensbi.experimental.models.autoencoders.commons) dim (gensbi.models.flux1.layers.EmbedND attribute) dim_cond (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.Flux1Params attribute) (gensbi.recipes.flux1.Flux1DiffusionPipeline attribute) (gensbi.recipes.flux1.Flux1FlowPipeline attribute) (gensbi.recipes.Flux1DiffusionPipeline attribute) (gensbi.recipes.Flux1FlowPipeline attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) dim_condition (gensbi.models.Simformer attribute) (gensbi.models.simformer.model.Simformer attribute) (gensbi.models.simformer.model.SimformerParams attribute) (gensbi.models.simformer.Simformer attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute)

dim_id (gensbi.models.Simformer attribute) (gensbi.models.simformer.model.Simformer attribute) (gensbi.models.simformer.model.SimformerParams attribute) (gensbi.models.simformer.Simformer attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute) dim_joint (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.simformer.model.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute) (gensbi.recipes.flux1joint.Flux1JointDiffusionPipeline attribute) (gensbi.recipes.flux1joint.Flux1JointFlowPipeline attribute) (gensbi.recipes.Flux1JointDiffusionPipeline attribute) (gensbi.recipes.Flux1JointFlowPipeline attribute) (gensbi.recipes.joint_pipeline.JointFlowPipeline attribute) (gensbi.recipes.JointFlowPipeline attribute) (gensbi.recipes.simformer.SimformerDiffusionPipeline attribute) (gensbi.recipes.simformer.SimformerFlowPipeline attribute) (gensbi.recipes.SimformerDiffusionPipeline attribute) (gensbi.recipes.SimformerFlowPipeline attribute) dim_obs (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.Flux1Params attribute) (gensbi.recipes.flux1.Flux1DiffusionPipeline attribute) (gensbi.recipes.flux1.Flux1FlowPipeline attribute) (gensbi.recipes.Flux1DiffusionPipeline attribute) (gensbi.recipes.Flux1FlowPipeline attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) dim_value (gensbi.models.Simformer attribute) (gensbi.models.simformer.model.Simformer attribute) (gensbi.models.simformer.model.SimformerParams attribute) (gensbi.models.simformer.Simformer attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute) din (gensbi.models.simformer.transformer.Transformer attribute) divergence() (in module gensbi.utils.math) double_blocks (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.model.Flux1 attribute) DoubleStreamBlock (class in gensbi.models.flux1.layers) down (gensbi.experimental.models.autoencoders.autoencoder_1d.Encoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Encoder2D attribute) Downsample1D (class in gensbi.experimental.models.autoencoders.autoencoder_1d) Downsample2D (class in gensbi.experimental.models.autoencoders.autoencoder_2d) dx_t (gensbi.flow_matching.path.path_sample.PathSample attribute), [1] (gensbi.flow_matching.path.PathSample attribute), [1]

E

e_s (gensbi.diffusion.path.scheduler.edm.VPScheduler attribute) (gensbi.diffusion.path.scheduler.VPScheduler attribute) e_t (gensbi.diffusion.path.scheduler.edm.VPScheduler attribute) (gensbi.diffusion.path.scheduler.VPScheduler attribute) edge_mask (gensbi.recipes.simformer.SimformerDiffusionPipeline attribute) (gensbi.recipes.simformer.SimformerFlowPipeline attribute) (gensbi.recipes.SimformerDiffusionPipeline attribute) (gensbi.recipes.SimformerFlowPipeline attribute) 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) ema_model (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline attribute) (gensbi.recipes.flux1.Flux1DiffusionPipeline attribute) (gensbi.recipes.flux1.Flux1FlowPipeline attribute) (gensbi.recipes.Flux1DiffusionPipeline attribute) (gensbi.recipes.Flux1FlowPipeline attribute) (gensbi.recipes.flux1joint.Flux1JointDiffusionPipeline attribute) (gensbi.recipes.flux1joint.Flux1JointFlowPipeline attribute) (gensbi.recipes.Flux1JointDiffusionPipeline attribute) (gensbi.recipes.Flux1JointFlowPipeline attribute) (gensbi.recipes.simformer.SimformerDiffusionPipeline attribute) (gensbi.recipes.simformer.SimformerFlowPipeline attribute) (gensbi.recipes.SimformerDiffusionPipeline attribute) (gensbi.recipes.SimformerFlowPipeline attribute) ema_model_wrapped (gensbi.recipes.pipeline.AbstractPipeline attribute) ema_step() (in module gensbi.recipes.pipeline) Embed (class in gensbi.models.embedding) (class in gensbi.models.embedding.embedding)

embed (gensbi.models.embedding.Embed attribute) (gensbi.models.embedding.embedding.Embed attribute) embedding_net_id (gensbi.models.Simformer attribute) (gensbi.models.simformer.model.Simformer attribute) (gensbi.models.simformer.Simformer attribute) embedding_time (gensbi.models.Simformer attribute) (gensbi.models.simformer.model.Simformer attribute) (gensbi.models.simformer.Simformer attribute) EmbedND (class in gensbi.models.flux1.layers) encode() (gensbi.experimental.models.AutoEncoder1D method) (gensbi.experimental.models.AutoEncoder2D method) (gensbi.experimental.models.autoencoders.AutoEncoder1D method) (gensbi.experimental.models.autoencoders.AutoEncoder2D method) (gensbi.experimental.models.autoencoders.autoencoder_1d.AutoEncoder1D method) (gensbi.experimental.models.autoencoders.autoencoder_2d.AutoEncoder2D method) Encoder1D (class in gensbi.experimental.models.autoencoders.autoencoder_1d) (gensbi.experimental.models.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.AutoEncoder1D attribute) Encoder2D (class in gensbi.experimental.models.autoencoders.autoencoder_2d) (gensbi.experimental.models.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AutoEncoder2D attribute) EnsembleClassifier (class in gensbi.diagnostics.lc2st) 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) eval_lc2st() (in module gensbi.diagnostics.lc2st) expand_tensor_like() (in module gensbi.flow_matching.utils) (in module gensbi.flow_matching.utils.utils)

F

f() (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.diffusion.path.scheduler.edm.VPScheduler method) (gensbi.diffusion.path.scheduler.VPScheduler method) FeatureEmbedder (class in gensbi.models.embedding) (class in gensbi.models.embedding.embedding) ffactor (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D attribute) final_layer (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.model.Flux1 attribute) (gensbi.models.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1Joint attribute) (gensbi.models.flux1joint.model.Flux1Joint attribute) fit() (gensbi.diagnostics.lc2st.EnsembleClassifier method) Flux1 (class in gensbi.models) (class in gensbi.models.flux1) (class in gensbi.models.flux1.model) Flux1DiffusionPipeline (class in gensbi.recipes) (class in gensbi.recipes.flux1)

Flux1FlowPipeline (class in gensbi.recipes) (class in gensbi.recipes.flux1) Flux1Joint (class in gensbi.models) (class in gensbi.models.flux1joint) (class in gensbi.models.flux1joint.model) Flux1JointDiffusionPipeline (class in gensbi.recipes) (class in gensbi.recipes.flux1joint) Flux1JointFlowPipeline (class in gensbi.recipes) (class in gensbi.recipes.flux1joint) Flux1JointParams (class in gensbi.models) (class in gensbi.models.flux1joint) (class in gensbi.models.flux1joint.model) Flux1Params (class in gensbi.models) (class in gensbi.models.flux1) (class in gensbi.models.flux1.model) fourier_features (gensbi.models.simformer.model.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.embedding) (class in gensbi.models.embedding.embedding) gensbi module gensbi.diagnostics module gensbi.diagnostics.distribution_wrapper module gensbi.diagnostics.lc2st module gensbi.diagnostics.metrics module gensbi.diagnostics.metrics.c2st module gensbi.diagnostics.metrics.l1_l2 module gensbi.diagnostics.sbc module gensbi.diagnostics.tarp 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.experimental module gensbi.experimental.models module gensbi.experimental.models.autoencoders module gensbi.experimental.models.autoencoders.autoencoder_1d module gensbi.experimental.models.autoencoders.autoencoder_2d module gensbi.experimental.models.autoencoders.commons module gensbi.experimental.recipes module gensbi.experimental.recipes.vae_pipeline 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.solver module gensbi.flow_matching.utils module gensbi.flow_matching.utils.utils module gensbi.models module gensbi.models.embedding module gensbi.models.embedding.embedding module gensbi.models.flux1 module gensbi.models.flux1.layers module gensbi.models.flux1.math module gensbi.models.flux1.model module gensbi.models.flux1joint module gensbi.models.flux1joint.model module gensbi.models.losses module gensbi.models.losses.conditional module

gensbi.models.losses.joint module gensbi.models.losses.unconditional module gensbi.models.simformer module gensbi.models.simformer.model module gensbi.models.simformer.transformer module gensbi.models.wrappers module gensbi.models.wrappers.conditional module gensbi.models.wrappers.joint module gensbi.models.wrappers.unconditional module gensbi.recipes module gensbi.recipes.conditional_pipeline module gensbi.recipes.flux1 module gensbi.recipes.flux1joint module gensbi.recipes.joint_pipeline module gensbi.recipes.pipeline module gensbi.recipes.simformer module gensbi.recipes.unconditional_pipeline module gensbi.recipes.utils module gensbi.utils module gensbi.utils.math module gensbi.utils.misc module gensbi.utils.model_wrapping module gensbi.utils.plotting module get_batch() (gensbi.diffusion.path.path_sample.EDMPathSample method) get_colored_value() (in module gensbi.utils.misc) get_default_training_config() (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline class method) (gensbi.recipes.conditional_pipeline.ConditionalDiffusionPipeline class method) (gensbi.recipes.ConditionalDiffusionPipeline class method) (gensbi.recipes.joint_pipeline.JointDiffusionPipeline class method) (gensbi.recipes.JointDiffusionPipeline class method) (gensbi.recipes.pipeline.AbstractPipeline class method) (gensbi.recipes.unconditional_pipeline.UnconditionalDiffusionPipeline class method) (gensbi.recipes.UnconditionalDiffusionPipeline class method) get_divergence() (gensbi.utils.model_wrapping.ModelWrapper method) get_loss_fn() (gensbi.diffusion.path.edm_path.EDMPath method) (gensbi.diffusion.path.EDMPath method) (gensbi.diffusion.path.scheduler.edm.BaseSDE method) (gensbi.recipes.conditional_pipeline.ConditionalDiffusionPipeline method) (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline method) (gensbi.recipes.ConditionalDiffusionPipeline method) (gensbi.recipes.ConditionalFlowPipeline method) (gensbi.recipes.joint_pipeline.JointDiffusionPipeline method) (gensbi.recipes.joint_pipeline.JointFlowPipeline method) (gensbi.recipes.JointDiffusionPipeline method) (gensbi.recipes.JointFlowPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalDiffusionPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalFlowPipeline method) (gensbi.recipes.UnconditionalDiffusionPipeline method) (gensbi.recipes.UnconditionalFlowPipeline method) 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.recipes.conditional_pipeline.ConditionalDiffusionPipeline method) (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline method) (gensbi.recipes.ConditionalDiffusionPipeline method) (gensbi.recipes.ConditionalFlowPipeline method) (gensbi.recipes.joint_pipeline.JointDiffusionPipeline method) (gensbi.recipes.joint_pipeline.JointFlowPipeline method) (gensbi.recipes.JointDiffusionPipeline method) (gensbi.recipes.JointFlowPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalDiffusionPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalFlowPipeline method) (gensbi.recipes.UnconditionalDiffusionPipeline method) (gensbi.recipes.UnconditionalFlowPipeline method) get_scores() (gensbi.diagnostics.LC2ST method) (gensbi.diagnostics.lc2st.LC2ST method) get_statistic_on_observed_data() (gensbi.diagnostics.LC2ST method) (gensbi.diagnostics.lc2st.LC2ST method) get_statistics_under_null_hypothesis() (gensbi.diagnostics.LC2ST method) (gensbi.diagnostics.lc2st.LC2ST method) get_tarp_references() (in module gensbi.diagnostics.tarp) get_train_step_fn() (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline method) (gensbi.recipes.conditional_pipeline.ConditionalDiffusionPipeline method) (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline method) (gensbi.recipes.ConditionalDiffusionPipeline method) (gensbi.recipes.ConditionalFlowPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) get_unnormalized_logprob() (gensbi.flow_matching.solver.ode_solver.ODESolver method) (gensbi.flow_matching.solver.ODESolver method) get_val_step_fn() (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) get_vector_field() (gensbi.utils.model_wrapping.ModelWrapper method) GREEN (in module gensbi.utils.misc) guidance_embed (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.Flux1Params attribute)

H

hidden_blocks (gensbi.models.simformer.transformer.DenseBlock attribute) hidden_dim (gensbi.models.flux1.layers.SingleStreamBlock attribute) hidden_size (gensbi.models.embedding.embedding.SinusoidalPosEmbed1D attribute) (gensbi.models.embedding.embedding.SinusoidalPosEmbed2D attribute) (gensbi.models.embedding.SinusoidalPosEmbed1D attribute) (gensbi.models.embedding.SinusoidalPosEmbed2D attribute) (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.layers.DoubleStreamBlock attribute) (gensbi.models.flux1.layers.SingleStreamBlock attribute) (gensbi.models.flux1.model.Flux1 attribute) (gensbi.models.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1Joint attribute) (gensbi.models.flux1joint.model.Flux1Joint attribute)

hist_color (in module gensbi.utils.plotting)

I

id_embedding_strategy (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.Flux1Params attribute) Identity (class in gensbi.models.flux1.layers) in_ch_mult (gensbi.experimental.models.autoencoders.autoencoder_1d.Encoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Encoder2D attribute) in_channels (gensbi.experimental.models.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.autoencoder_1d.AttnBlock1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.Encoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.ResnetBlock1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AttnBlock2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Encoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.ResnetBlock2D attribute) (gensbi.experimental.models.autoencoders.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.commons.AutoEncoderParams attribute), [1] (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1 attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1Joint attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.Flux1Params attribute) (gensbi.models.Simformer attribute) (gensbi.models.simformer.model.Simformer attribute) (gensbi.models.simformer.model.SimformerParams attribute) (gensbi.models.simformer.Simformer attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute)

in_layer (gensbi.models.embedding.embedding.MLPEmbedder attribute) (gensbi.models.embedding.MLPEmbedder attribute) (gensbi.models.flux1.layers.MLPEmbedder attribute) init_ids_1d() (in module gensbi.recipes.utils) init_ids_2d() (in module gensbi.recipes.utils) init_ids_joint() (in module gensbi.recipes.utils) init_pipeline_from_config() (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline method) (gensbi.recipes.conditional_pipeline.ConditionalDiffusionPipeline class method) (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline class method) (gensbi.recipes.ConditionalDiffusionPipeline class method) (gensbi.recipes.ConditionalFlowPipeline class method) (gensbi.recipes.flux1.Flux1DiffusionPipeline class method) (gensbi.recipes.flux1.Flux1FlowPipeline class method) (gensbi.recipes.Flux1DiffusionPipeline class method) (gensbi.recipes.Flux1FlowPipeline class method) (gensbi.recipes.flux1joint.Flux1JointDiffusionPipeline class method) (gensbi.recipes.flux1joint.Flux1JointFlowPipeline class method) (gensbi.recipes.Flux1JointDiffusionPipeline class method) (gensbi.recipes.Flux1JointFlowPipeline class method) (gensbi.recipes.joint_pipeline.JointDiffusionPipeline class method) (gensbi.recipes.joint_pipeline.JointFlowPipeline class method) (gensbi.recipes.JointDiffusionPipeline class method) (gensbi.recipes.JointFlowPipeline class method) (gensbi.recipes.pipeline.AbstractPipeline method) (gensbi.recipes.simformer.SimformerDiffusionPipeline class method) (gensbi.recipes.simformer.SimformerFlowPipeline class method) (gensbi.recipes.SimformerDiffusionPipeline class method) (gensbi.recipes.SimformerFlowPipeline class method) (gensbi.recipes.unconditional_pipeline.UnconditionalDiffusionPipeline class method) (gensbi.recipes.unconditional_pipeline.UnconditionalFlowPipeline class method) (gensbi.recipes.UnconditionalDiffusionPipeline class method) (gensbi.recipes.UnconditionalFlowPipeline class method) is_double (gensbi.models.flux1.layers.Modulation attribute)

J

JointCFMLoss (class in gensbi.models) (class in gensbi.models.losses) (class in gensbi.models.losses.joint) JointDiffLoss (class in gensbi.models) (class in gensbi.models.losses) (class in gensbi.models.losses.joint)

JointDiffusionPipeline (class in gensbi.recipes) (class in gensbi.recipes.joint_pipeline) JointFlowPipeline (class in gensbi.recipes) (class in gensbi.recipes.joint_pipeline) JointWrapper (class in gensbi.models) (class in gensbi.models.wrappers) (class in gensbi.models.wrappers.joint)

K

k (gensbi.experimental.models.autoencoders.autoencoder_1d.AttnBlock1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AttnBlock2D attribute) 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)

key_norm (gensbi.models.flux1.layers.QKNorm attribute) kl_loss (gensbi.experimental.models.autoencoders.commons.DiagonalGaussian attribute) kwargs (gensbi.diagnostics.distribution_wrapper.PosteriorWrapper attribute) (gensbi.diagnostics.PosteriorWrapper attribute)

L

l1() (in module gensbi.diagnostics.metrics) (in module gensbi.diagnostics.metrics.l1_l2) l2() (in module gensbi.diagnostics.metrics) (in module gensbi.diagnostics.metrics.l1_l2) LastLayer (class in gensbi.models.flux1.layers) latent_shape (gensbi.experimental.models.AutoEncoder1D attribute) (gensbi.experimental.models.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AutoEncoder2D attribute) layer_norm (gensbi.models.simformer.transformer.AttentionBlock attribute) (gensbi.models.simformer.transformer.DenseBlock attribute) (gensbi.models.simformer.transformer.Transformer attribute) LC2ST (class in gensbi.diagnostics) (class in gensbi.diagnostics.lc2st) lin (gensbi.models.flux1.layers.Modulation attribute) linear (gensbi.models.flux1.layers.LastLayer attribute) linear1 (gensbi.models.flux1.layers.SingleStreamBlock attribute) linear2 (gensbi.models.flux1.layers.SingleStreamBlock attribute) LinearVPScheduler (class in gensbi.flow_matching.path.scheduler) (class in gensbi.flow_matching.path.scheduler.scheduler) Loss (class in gensbi.experimental.models.autoencoders.commons) loss_fn (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline attribute) (gensbi.models.ConditionalDiffLoss attribute) (gensbi.models.JointDiffLoss attribute) (gensbi.models.losses.conditional.ConditionalDiffLoss attribute) (gensbi.models.losses.ConditionalDiffLoss attribute) (gensbi.models.losses.joint.JointDiffLoss attribute) (gensbi.models.losses.JointDiffLoss attribute) (gensbi.models.losses.unconditional.UnconditionalDiffLoss attribute) (gensbi.models.losses.UnconditionalDiffLoss attribute) (gensbi.models.UnconditionalDiffLoss attribute) (gensbi.recipes.conditional_pipeline.ConditionalDiffusionPipeline attribute) (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline attribute) (gensbi.recipes.ConditionalDiffusionPipeline attribute) (gensbi.recipes.ConditionalFlowPipeline attribute) (gensbi.recipes.joint_pipeline.JointDiffusionPipeline attribute) (gensbi.recipes.joint_pipeline.JointFlowPipeline attribute) (gensbi.recipes.JointDiffusionPipeline attribute) (gensbi.recipes.JointFlowPipeline attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) (gensbi.recipes.unconditional_pipeline.UnconditionalDiffusionPipeline attribute) (gensbi.recipes.unconditional_pipeline.UnconditionalFlowPipeline attribute) (gensbi.recipes.UnconditionalDiffusionPipeline attribute) (gensbi.recipes.UnconditionalFlowPipeline attribute)

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

M (gensbi.diffusion.path.scheduler.edm.VPScheduler attribute) (gensbi.diffusion.path.scheduler.VPScheduler attribute) mid (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.Encoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Encoder2D attribute) mlp_act (gensbi.models.flux1.layers.SingleStreamBlock attribute) mlp_hidden_dim (gensbi.models.flux1.layers.SingleStreamBlock attribute) mlp_ratio (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.Flux1Params attribute) MLPEmbedder (class in gensbi.models.embedding) (class in gensbi.models.embedding.embedding) (class in gensbi.models.flux1.layers) model (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) (gensbi.utils.model_wrapping.ModelWrapper attribute) model_wrapped (gensbi.recipes.pipeline.AbstractPipeline attribute) ModelEMA (class in gensbi.recipes.pipeline) ModelWrapper (class in gensbi.utils.model_wrapping) Modulation (class in gensbi.models.flux1.layers) modulation (gensbi.models.flux1.layers.SingleStreamBlock attribute) ModulationOut (class in gensbi.models.flux1.layers) module gensbi gensbi.diagnostics gensbi.diagnostics.distribution_wrapper gensbi.diagnostics.lc2st gensbi.diagnostics.metrics gensbi.diagnostics.metrics.c2st gensbi.diagnostics.metrics.l1_l2 gensbi.diagnostics.sbc gensbi.diagnostics.tarp 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.experimental gensbi.experimental.models gensbi.experimental.models.autoencoders gensbi.experimental.models.autoencoders.autoencoder_1d gensbi.experimental.models.autoencoders.autoencoder_2d gensbi.experimental.models.autoencoders.commons gensbi.experimental.recipes gensbi.experimental.recipes.vae_pipeline 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.solver gensbi.flow_matching.utils gensbi.flow_matching.utils.utils gensbi.models gensbi.models.embedding gensbi.models.embedding.embedding gensbi.models.flux1 gensbi.models.flux1.layers gensbi.models.flux1.math gensbi.models.flux1.model gensbi.models.flux1joint gensbi.models.flux1joint.model gensbi.models.losses gensbi.models.losses.conditional gensbi.models.losses.joint gensbi.models.losses.unconditional gensbi.models.simformer gensbi.models.simformer.model gensbi.models.simformer.transformer gensbi.models.wrappers gensbi.models.wrappers.conditional gensbi.models.wrappers.joint gensbi.models.wrappers.unconditional gensbi.recipes gensbi.recipes.conditional_pipeline gensbi.recipes.flux1 gensbi.recipes.flux1joint gensbi.recipes.joint_pipeline gensbi.recipes.pipeline gensbi.recipes.simformer gensbi.recipes.unconditional_pipeline gensbi.recipes.utils gensbi.utils gensbi.utils.math gensbi.utils.misc gensbi.utils.model_wrapping gensbi.utils.plotting

multiplier (gensbi.models.flux1.layers.Modulation attribute)

N

n (gensbi.flow_matching.path.scheduler.PolynomialConvexScheduler attribute) (gensbi.flow_matching.path.scheduler.scheduler.PolynomialConvexScheduler attribute) 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) new_scheduler (gensbi.flow_matching.path.scheduler.schedule_transform.ScheduleTransformedModel attribute) (gensbi.flow_matching.path.scheduler.ScheduleTransformedModel attribute) norm (gensbi.experimental.models.autoencoders.autoencoder_1d.AttnBlock1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AttnBlock2D attribute) (gensbi.models.flux1.layers.SelfAttention attribute) (gensbi.models.flux1.layers.SingleStreamBlock attribute) norm1 (gensbi.experimental.models.autoencoders.autoencoder_1d.ResnetBlock1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.ResnetBlock2D attribute) norm2 (gensbi.experimental.models.autoencoders.autoencoder_1d.ResnetBlock1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.ResnetBlock2D attribute) norm_final (gensbi.models.flux1.layers.LastLayer attribute) norm_out (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.Encoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Encoder2D attribute) null_distribution (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) num_ensemble (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.EnsembleClassifier attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) num_folds (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) num_heads (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.layers.DoubleStreamBlock attribute) (gensbi.models.flux1.layers.SelfAttention attribute) (gensbi.models.flux1.layers.SingleStreamBlock attribute) (gensbi.models.flux1.model.Flux1 attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1Joint attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.Flux1Params attribute) (gensbi.models.simformer.model.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.simformer.transformer.Transformer attribute) (gensbi.models.SimformerParams attribute)

num_hidden_layers (gensbi.models.simformer.model.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.simformer.transformer.Transformer attribute) (gensbi.models.SimformerParams attribute) num_layers (gensbi.models.simformer.model.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.simformer.transformer.Transformer attribute) (gensbi.models.SimformerParams attribute) num_res_blocks (gensbi.experimental.models.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.Encoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Encoder2D attribute) (gensbi.experimental.models.autoencoders.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.commons.AutoEncoderParams attribute), [1] num_resolutions (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.Encoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Encoder2D attribute) num_trials_null (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute)

O

obs_attn (gensbi.models.flux1.layers.DoubleStreamBlock attribute) obs_ids (gensbi.recipes.conditional_pipeline.ConditionalDiffusionPipeline attribute) (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline attribute) (gensbi.recipes.ConditionalDiffusionPipeline attribute) (gensbi.recipes.ConditionalFlowPipeline attribute) (gensbi.recipes.unconditional_pipeline.UnconditionalDiffusionPipeline attribute) (gensbi.recipes.unconditional_pipeline.UnconditionalFlowPipeline attribute) (gensbi.recipes.UnconditionalDiffusionPipeline attribute) (gensbi.recipes.UnconditionalFlowPipeline attribute) obs_in (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.model.Flux1 attribute) (gensbi.models.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1Joint attribute) (gensbi.models.flux1joint.model.Flux1Joint attribute) obs_mlp (gensbi.models.flux1.layers.DoubleStreamBlock attribute) obs_mod (gensbi.models.flux1.layers.DoubleStreamBlock attribute) obs_norm1 (gensbi.models.flux1.layers.DoubleStreamBlock attribute) obs_norm2 (gensbi.models.flux1.layers.DoubleStreamBlock attribute) ODESolver (class in gensbi.flow_matching.solver) (class in gensbi.flow_matching.solver.ode_solver) original_scheduler (gensbi.flow_matching.path.scheduler.schedule_transform.ScheduleTransformedModel attribute) (gensbi.flow_matching.path.scheduler.ScheduleTransformedModel attribute)

out_ch (gensbi.experimental.models.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.commons.AutoEncoderParams attribute), [1] out_channels (gensbi.experimental.models.autoencoders.autoencoder_1d.ResnetBlock1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.ResnetBlock2D attribute) (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.model.Flux1 attribute) (gensbi.models.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1Joint attribute) (gensbi.models.flux1joint.model.Flux1Joint attribute) out_layer (gensbi.models.embedding.embedding.MLPEmbedder attribute) (gensbi.models.embedding.MLPEmbedder attribute) (gensbi.models.flux1.layers.MLPEmbedder attribute) output_dim (gensbi.models.embedding.embedding.GaussianFourierEmbedding attribute) (gensbi.models.embedding.embedding.SinusoidalTimeEmbedding attribute) (gensbi.models.embedding.GaussianFourierEmbedding attribute) (gensbi.models.embedding.SinusoidalTimeEmbedding attribute) output_fn (gensbi.models.Simformer attribute) (gensbi.models.simformer.model.Simformer attribute) (gensbi.models.simformer.Simformer attribute)

P

p0_dist_model (gensbi.recipes.pipeline.AbstractPipeline attribute) p0_joint (gensbi.recipes.joint_pipeline.JointFlowPipeline attribute) (gensbi.recipes.JointFlowPipeline attribute) p0_obs (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline attribute) (gensbi.recipes.ConditionalFlowPipeline attribute) (gensbi.recipes.joint_pipeline.JointFlowPipeline attribute) (gensbi.recipes.JointFlowPipeline attribute) (gensbi.recipes.unconditional_pipeline.UnconditionalFlowPipeline attribute) (gensbi.recipes.UnconditionalFlowPipeline attribute) P_mean (gensbi.diffusion.path.scheduler.edm.EDMScheduler attribute) (gensbi.diffusion.path.scheduler.EDMScheduler attribute) p_skip (gensbi.models.embedding.embedding.MLPEmbedder attribute) (gensbi.models.embedding.MLPEmbedder attribute) P_std (gensbi.diffusion.path.scheduler.edm.EDMScheduler attribute) (gensbi.diffusion.path.scheduler.EDMScheduler attribute) p_value() (gensbi.diagnostics.LC2ST method) (gensbi.diagnostics.lc2st.LC2ST method) param_dtype (gensbi.experimental.models.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.commons.AutoEncoderParams attribute), [1] (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.Flux1Params attribute) (gensbi.models.simformer.model.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.simformer.transformer.Transformer attribute) (gensbi.models.SimformerParams attribute) params (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline attribute) (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.model.Flux1 attribute) (gensbi.models.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1Joint attribute) (gensbi.models.flux1joint.model.Flux1Joint attribute) (gensbi.models.Simformer attribute) (gensbi.models.simformer.model.Simformer attribute) (gensbi.models.simformer.Simformer attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) parse_autoencoder_params() (in module gensbi.experimental.recipes.vae_pipeline) parse_flux1_params() (in module gensbi.recipes.flux1) parse_flux1joint_params() (in module gensbi.recipes.flux1joint) parse_simformer_params() (in module gensbi.recipes.simformer) parse_training_config() (in module gensbi.experimental.recipes.vae_pipeline) (in module gensbi.recipes.flux1) (in module gensbi.recipes.flux1joint) (in module gensbi.recipes.simformer) patchify_2d() (in module gensbi.recipes.utils) path (gensbi.diffusion.solver.sde_solver.SDESolver attribute) (gensbi.diffusion.solver.SDESolver attribute) (gensbi.flow_matching.loss.continuous_loss.ContinuousFMLoss attribute) (gensbi.flow_matching.loss.ContinuousFMLoss attribute) (gensbi.models.ConditionalDiffLoss attribute) (gensbi.models.JointDiffLoss attribute) (gensbi.models.losses.conditional.ConditionalDiffLoss attribute) (gensbi.models.losses.ConditionalDiffLoss attribute) (gensbi.models.losses.joint.JointDiffLoss attribute) (gensbi.models.losses.JointDiffLoss attribute) (gensbi.models.losses.unconditional.UnconditionalDiffLoss attribute) (gensbi.models.losses.UnconditionalDiffLoss attribute) (gensbi.models.UnconditionalDiffLoss attribute) (gensbi.recipes.conditional_pipeline.ConditionalDiffusionPipeline attribute) (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline attribute) (gensbi.recipes.ConditionalDiffusionPipeline attribute) (gensbi.recipes.ConditionalFlowPipeline attribute) (gensbi.recipes.joint_pipeline.JointDiffusionPipeline attribute) (gensbi.recipes.joint_pipeline.JointFlowPipeline attribute) (gensbi.recipes.JointDiffusionPipeline attribute) (gensbi.recipes.JointFlowPipeline attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) (gensbi.recipes.unconditional_pipeline.UnconditionalDiffusionPipeline attribute) (gensbi.recipes.unconditional_pipeline.UnconditionalFlowPipeline attribute) (gensbi.recipes.UnconditionalDiffusionPipeline attribute) (gensbi.recipes.UnconditionalFlowPipeline attribute)

PathSample (class in gensbi.flow_matching.path) (class in gensbi.flow_matching.path.path_sample) pe (gensbi.models.embedding.embedding.SinusoidalPosEmbed1D attribute) (gensbi.models.embedding.SinusoidalPosEmbed1D attribute) pe_h (gensbi.models.embedding.embedding.SinusoidalPosEmbed2D attribute) (gensbi.models.embedding.SinusoidalPosEmbed2D attribute) pe_w (gensbi.models.embedding.embedding.SinusoidalPosEmbed2D attribute) (gensbi.models.embedding.SinusoidalPosEmbed2D attribute) PEMatrix (class in gensbi.models.embedding.embedding) permutation (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) permute_data() (in module gensbi.diagnostics.lc2st) pipeline (gensbi.diagnostics.distribution_wrapper.PosteriorWrapper attribute) (gensbi.diagnostics.PosteriorWrapper attribute) plot_2d_dist_contour() (in module gensbi.utils.plotting) plot_2d_levels() (in module gensbi.utils.plotting) plot_lc2st() (in module gensbi.diagnostics) (in module gensbi.diagnostics.lc2st) plot_marginals() (in module gensbi.utils.plotting) plot_tarp() (in module gensbi.diagnostics) (in module gensbi.diagnostics.tarp) plot_trajectories() (in module gensbi.utils.plotting) PolynomialConvexScheduler (class in gensbi.flow_matching.path.scheduler) (class in gensbi.flow_matching.path.scheduler.scheduler) PosteriorWrapper (class in gensbi.diagnostics) (class in gensbi.diagnostics.distribution_wrapper) pre_norm (gensbi.models.flux1.layers.SingleStreamBlock attribute) predict_proba() (gensbi.diagnostics.lc2st.EnsembleClassifier method) ProbPath (class in gensbi.diffusion.path.path) (class in gensbi.flow_matching.path) (class in gensbi.flow_matching.path.path) proj (gensbi.models.flux1.layers.SelfAttention attribute) proj_out (gensbi.experimental.models.autoencoders.autoencoder_1d.AttnBlock1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AttnBlock2D attribute)

Q

q (gensbi.experimental.models.autoencoders.autoencoder_1d.AttnBlock1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AttnBlock2D attribute) QKNorm (class in gensbi.models.flux1.layers) qkv (gensbi.models.flux1.layers.SelfAttention attribute) qkv_bias (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.Flux1Params attribute)

qkv_features (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.layers.DoubleStreamBlock attribute) (gensbi.models.flux1.model.Flux1 attribute) (gensbi.models.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1Joint attribute) (gensbi.models.flux1joint.model.Flux1Joint attribute) (gensbi.models.simformer.model.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.SimformerParams attribute) query_norm (gensbi.models.flux1.layers.QKNorm attribute)

R

RED (in module gensbi.utils.misc) reg (gensbi.experimental.models.AutoEncoder1D attribute) (gensbi.experimental.models.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AutoEncoder2D attribute) reject_test() (gensbi.diagnostics.LC2ST method) (gensbi.diagnostics.lc2st.LC2ST method) repeats (gensbi.models.embedding.embedding.MLPEmbedder attribute) (gensbi.models.embedding.MLPEmbedder attribute) RESET (in module gensbi.utils.misc) ResnetBlock1D (class in gensbi.experimental.models.autoencoders.autoencoder_1d) ResnetBlock2D (class in gensbi.experimental.models.autoencoders.autoencoder_2d) resolution (gensbi.experimental.models.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.Encoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Encoder2D attribute) (gensbi.experimental.models.autoencoders.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.commons.AutoEncoderParams attribute), [1] restore_model() (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) rgb_base (in module gensbi.utils.plotting) rho (gensbi.diffusion.path.scheduler.edm.EDMScheduler attribute) (gensbi.diffusion.path.scheduler.EDMScheduler attribute) rngs (gensbi.diagnostics.distribution_wrapper.PosteriorWrapper attribute) (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) (gensbi.diagnostics.PosteriorWrapper attribute) (gensbi.experimental.models.AutoEncoder1D attribute) (gensbi.experimental.models.AutoEncoder2D attribute) (gensbi.experimental.models.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.commons.AutoEncoderParams attribute), [1] (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.Flux1Params attribute) (gensbi.models.simformer.model.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.simformer.transformer.Transformer attribute) (gensbi.models.SimformerParams attribute)

rope() (in module gensbi.models.flux1.math) run_sbc() (in module gensbi.diagnostics) (in module gensbi.diagnostics.sbc) run_tarp() (in module gensbi.diagnostics) (in module gensbi.diagnostics.tarp)

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.experimental.models.autoencoders.commons.DiagonalGaussian attribute) sample() (gensbi.diagnostics.distribution_wrapper.PosteriorWrapper method) (gensbi.diagnostics.PosteriorWrapper method) (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.Solver method) (gensbi.flow_matching.solver.solver.Solver method) (gensbi.recipes.conditional_pipeline.ConditionalDiffusionPipeline method) (gensbi.recipes.conditional_pipeline.ConditionalFlowPipeline method) (gensbi.recipes.ConditionalDiffusionPipeline method) (gensbi.recipes.ConditionalFlowPipeline method) (gensbi.recipes.joint_pipeline.JointDiffusionPipeline method) (gensbi.recipes.joint_pipeline.JointFlowPipeline method) (gensbi.recipes.JointDiffusionPipeline method) (gensbi.recipes.JointFlowPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) (gensbi.recipes.simformer.SimformerDiffusionPipeline method) (gensbi.recipes.simformer.SimformerFlowPipeline method) (gensbi.recipes.SimformerDiffusionPipeline method) (gensbi.recipes.SimformerFlowPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalDiffusionPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalFlowPipeline method) (gensbi.recipes.UnconditionalDiffusionPipeline method) (gensbi.recipes.UnconditionalFlowPipeline method) sample_batched() (gensbi.diagnostics.distribution_wrapper.PosteriorWrapper method) (gensbi.diagnostics.PosteriorWrapper method) (gensbi.recipes.pipeline.AbstractPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalDiffusionPipeline method) (gensbi.recipes.unconditional_pipeline.UnconditionalFlowPipeline method) (gensbi.recipes.UnconditionalDiffusionPipeline method) (gensbi.recipes.UnconditionalFlowPipeline method) sample_condition_mask() (in module gensbi.recipes.joint_pipeline) 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) sample_structured_conditional_mask() (in module gensbi.recipes.joint_pipeline) save_model() (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) sbc_rank_plot() (in module gensbi.diagnostics) (in module gensbi.diagnostics.sbc) scale (gensbi.models.flux1.layers.ModulationOut attribute) (gensbi.models.flux1.layers.SingleStreamBlock attribute) scale_factor (gensbi.experimental.models.AutoEncoder1D attribute) (gensbi.experimental.models.AutoEncoder2D attribute) (gensbi.experimental.models.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.commons.AutoEncoderParams attribute), [1] scale_lr() (in module gensbi.utils.misc) Scheduler (class in gensbi.flow_matching.path.scheduler) (class in gensbi.flow_matching.path.scheduler.scheduler) scheduler (gensbi.diffusion.path.edm_path.EDMPath attribute) (gensbi.diffusion.path.EDMPath attribute) (gensbi.diffusion.path.path.ProbPath attribute) (gensbi.flow_matching.path.affine.AffineProbPath attribute) (gensbi.flow_matching.path.affine.CondOTProbPath attribute) (gensbi.flow_matching.path.AffineProbPath attribute) (gensbi.flow_matching.path.CondOTProbPath attribute) 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)

score_model (gensbi.diffusion.solver.sde_solver.SDESolver attribute) (gensbi.diffusion.solver.SDESolver attribute) SDESolver (class in gensbi.diffusion.solver) (class in gensbi.diffusion.solver.sde_solver) seed (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) SelfAttention (class in gensbi.models.flux1.layers) set_default_x() (gensbi.diagnostics.distribution_wrapper.PosteriorWrapper method) (gensbi.diagnostics.PosteriorWrapper method) shift (gensbi.models.flux1.layers.ModulationOut attribute) shift_factor (gensbi.experimental.models.AutoEncoder1D attribute) (gensbi.experimental.models.AutoEncoder2D attribute) (gensbi.experimental.models.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.autoencoder_1d.AutoEncoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AutoEncoder2D attribute) (gensbi.experimental.models.autoencoders.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.commons.AutoEncoderParams attribute), [1] 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_data (gensbi.diffusion.path.scheduler.edm.EDMScheduler attribute) (gensbi.diffusion.path.scheduler.EDMScheduler attribute) 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_max (gensbi.diffusion.path.scheduler.edm.EDMScheduler attribute) (gensbi.diffusion.path.scheduler.edm.VEScheduler attribute) (gensbi.diffusion.path.scheduler.EDMScheduler attribute) (gensbi.diffusion.path.scheduler.VEScheduler attribute) sigma_min (gensbi.diffusion.path.scheduler.edm.EDMScheduler attribute) (gensbi.diffusion.path.scheduler.edm.VEScheduler attribute) (gensbi.diffusion.path.scheduler.EDMScheduler attribute) (gensbi.diffusion.path.scheduler.VEScheduler attribute) sigma_t (gensbi.flow_matching.path.scheduler.scheduler.SchedulerOutput attribute), [1] (gensbi.flow_matching.path.scheduler.SchedulerOutput attribute), [1] silu (gensbi.models.embedding.embedding.MLPEmbedder attribute) (gensbi.models.embedding.MLPEmbedder attribute) (gensbi.models.flux1.layers.MLPEmbedder attribute) Simformer (class in gensbi.models) (class in gensbi.models.simformer) (class in gensbi.models.simformer.model) SimformerDiffusionPipeline (class in gensbi.recipes) (class in gensbi.recipes.simformer) SimformerFlowPipeline (class in gensbi.recipes) (class in gensbi.recipes.simformer) SimformerParams (class in gensbi.models) (class in gensbi.models.simformer) (class in gensbi.models.simformer.model) SimpleTimeEmbedding (class in gensbi.models.embedding) (class in gensbi.models.embedding.embedding) single_blocks (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.model.Flux1 attribute) (gensbi.models.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1Joint attribute) (gensbi.models.flux1joint.model.Flux1Joint attribute) SingleStreamBlock (class in gensbi.models.flux1.layers) SinusoidalPosEmbed1D (class in gensbi.models.embedding) (class in gensbi.models.embedding.embedding) SinusoidalPosEmbed2D (class in gensbi.models.embedding) (class in gensbi.models.embedding.embedding) SinusoidalTimeEmbedding (class in gensbi.models.embedding) (class in gensbi.models.embedding.embedding) skip_connection (gensbi.models.simformer.transformer.AttentionBlock attribute) (gensbi.models.simformer.transformer.DenseBlock attribute) skip_connection_attn (gensbi.models.simformer.transformer.Transformer attribute) skip_connection_mlp (gensbi.models.simformer.transformer.Transformer attribute) 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.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.Flux1Params attribute) (gensbi.models.flux1.layers.EmbedND attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.Flux1Params attribute) theta_p (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) theta_p_mean (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) theta_p_std (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) theta_q (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) time_in (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.model.Flux1 attribute) (gensbi.models.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1Joint attribute) (gensbi.models.flux1joint.model.Flux1Joint 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) total_tokens (gensbi.models.Simformer attribute) (gensbi.models.simformer.model.Simformer attribute) (gensbi.models.simformer.Simformer attribute) train() (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) train_dataset (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) train_dataset_iter (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) train_lc2st() (in module gensbi.diagnostics.lc2st) train_on_observed_data() (gensbi.diagnostics.LC2ST method) (gensbi.diagnostics.lc2st.LC2ST method) train_under_null_hypothesis() (gensbi.diagnostics.LC2ST method) (gensbi.diagnostics.lc2st.LC2ST method) trained_clfs (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.EnsembleClassifier attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) trained_clfs_null (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) training_config (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) Transformer (class in gensbi.models.simformer.transformer) transformer (gensbi.models.Simformer attribute) (gensbi.models.simformer.model.Simformer attribute) (gensbi.models.simformer.Simformer attribute) transparent_cmap (in module gensbi.utils.plotting) true_val_color (in module gensbi.utils.plotting)

U

UnconditionalCFMLoss (class in gensbi.models) (class in gensbi.models.losses) (class in gensbi.models.losses.unconditional) UnconditionalDiffLoss (class in gensbi.models) (class in gensbi.models.losses) (class in gensbi.models.losses.unconditional) UnconditionalDiffusionPipeline (class in gensbi.recipes) (class in gensbi.recipes.unconditional_pipeline) UnconditionalFlowPipeline (class in gensbi.recipes) (class in gensbi.recipes.unconditional_pipeline) UnconditionalWrapper (class in gensbi.models) (class in gensbi.models.wrappers) (class in gensbi.models.wrappers.unconditional) unconditioned() (gensbi.models.JointWrapper method) (gensbi.models.wrappers.joint.JointWrapper method) (gensbi.models.wrappers.JointWrapper 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) up (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D attribute) update() (gensbi.recipes.pipeline.ModelEMA method) update_KL (gensbi.experimental.models.autoencoders.commons.DiagonalGaussian attribute) update_training_config() (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline method) (gensbi.recipes.pipeline.AbstractPipeline method) Upsample1D (class in gensbi.experimental.models.autoencoders.autoencoder_1d) Upsample2D (class in gensbi.experimental.models.autoencoders.autoencoder_2d)

V

v (gensbi.experimental.models.autoencoders.autoencoder_1d.AttnBlock1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.AttnBlock2D attribute) VAE1DPipeline (class in gensbi.experimental.recipes) (class in gensbi.experimental.recipes.vae_pipeline) VAE2DPipeline (class in gensbi.experimental.recipes) (class in gensbi.experimental.recipes.vae_pipeline) vae_loss_fn() (in module gensbi.experimental.models) (in module gensbi.experimental.models.autoencoders) (in module gensbi.experimental.models.autoencoders.commons) val_dataset (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) val_dataset_iter (gensbi.experimental.recipes.vae_pipeline.AbstractVAEPipeline attribute) (gensbi.recipes.pipeline.AbstractPipeline attribute) vec_in_dim (gensbi.models.flux1.Flux1Params attribute) (gensbi.models.flux1.model.Flux1Params attribute) (gensbi.models.flux1joint.Flux1JointParams attribute) (gensbi.models.flux1joint.model.Flux1JointParams attribute) (gensbi.models.Flux1JointParams attribute) (gensbi.models.Flux1Params attribute)

vector_in (gensbi.models.Flux1 attribute) (gensbi.models.flux1.Flux1 attribute) (gensbi.models.flux1.model.Flux1 attribute) (gensbi.models.Flux1Joint attribute) (gensbi.models.flux1joint.Flux1Joint attribute) (gensbi.models.flux1joint.model.Flux1Joint attribute) velocity_model (gensbi.flow_matching.solver.ode_solver.ODESolver attribute) (gensbi.flow_matching.solver.ODESolver 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) verbosity (gensbi.diagnostics.lc2st.EnsembleClassifier attribute) 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.model.SimformerParams attribute) (gensbi.models.simformer.SimformerParams attribute) (gensbi.models.simformer.transformer.Transformer attribute) (gensbi.models.SimformerParams attribute)

X

x_0 (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.PathSample attribute), [1] (gensbi.flow_matching.path.PathSample attribute), [1] x_p (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) x_p_mean (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute)

x_p_std (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) x_q (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) x_t (gensbi.diffusion.path.path_sample.EDMPathSample attribute), [1] (gensbi.flow_matching.path.path_sample.PathSample attribute), [1] (gensbi.flow_matching.path.PathSample attribute), [1]

Y

YELLOW (in module gensbi.utils.misc)

Z

z_channels (gensbi.experimental.models.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.AutoEncoderParams attribute), [1] (gensbi.experimental.models.autoencoders.commons.AutoEncoderParams attribute), [1]

z_score (gensbi.diagnostics.LC2ST attribute) (gensbi.diagnostics.lc2st.LC2ST attribute) z_shape (gensbi.experimental.models.autoencoders.autoencoder_1d.Decoder1D attribute) (gensbi.experimental.models.autoencoders.autoencoder_2d.Decoder2D attribute)