gensbi package

Subpackages

• gensbi.diffusion package
  • Subpackages
    • gensbi.diffusion.path package
      • Subpackages
      • Submodules
      • gensbi.diffusion.path.edm_path module
      • gensbi.diffusion.path.path module
      • gensbi.diffusion.path.path_sample module
      • Module contents
    • gensbi.diffusion.solver package
      • Submodules
      • gensbi.diffusion.solver.edm_samplers module
      • gensbi.diffusion.solver.sde_solver module
      • gensbi.diffusion.solver.solver module
      • Module contents
  • Module contents
• gensbi.flow_matching package
  • Subpackages
    • gensbi.flow_matching.loss package
      • Submodules
      • gensbi.flow_matching.loss.continuous_loss module
      • Module contents
    • gensbi.flow_matching.path package
      • Subpackages
      • Submodules
      • gensbi.flow_matching.path.affine module
      • gensbi.flow_matching.path.path module
      • gensbi.flow_matching.path.path_sample module
      • Module contents
    • gensbi.flow_matching.solver package
      • Submodules
      • gensbi.flow_matching.solver.ode_solver module
      • gensbi.flow_matching.solver.sde_solver module
      • gensbi.flow_matching.solver.solver module
      • gensbi.flow_matching.solver.utils module
      • Module contents
    • gensbi.flow_matching.utils package
      • Subpackages
      • Submodules
      • gensbi.flow_matching.utils.utils module
      • Module contents
  • Module contents
• gensbi.models package
  • Subpackages
    • gensbi.models.flux1 package
      • Submodules
      • gensbi.models.flux1.layers module
      • gensbi.models.flux1.loss module
      • gensbi.models.flux1.math module
      • gensbi.models.flux1.model module
      • Module contents
    • gensbi.models.simformer package
      • Submodules
      • gensbi.models.simformer.embedding module
      • gensbi.models.simformer.loss module
      • gensbi.models.simformer.simformer module
      • gensbi.models.simformer.transformer module
      • Module contents
  • Module contents
    • Flux
    • FluxCFMLoss
    • FluxParams
      • FluxParams.axes_dim
      • FluxParams.cond_dim
      • FluxParams.context_in_dim
      • FluxParams.depth
      • FluxParams.depth_single_blocks
      • FluxParams.guidance_embed
      • FluxParams.in_channels
      • FluxParams.mlp_ratio
      • FluxParams.num_heads
      • FluxParams.obs_dim
      • FluxParams.param_dtype
      • FluxParams.qkv_bias
      • FluxParams.qkv_multiplier
      • FluxParams.rngs
      • FluxParams.theta
      • FluxParams.use_rope
      • FluxParams.vec_in_dim
    • Simformer
    • SimformerCFMLoss
    • SimformerConditioner
      • SimformerConditioner.conditioned()
      • SimformerConditioner.unconditioned()
    • SimformerParams
      • SimformerParams.dim_condition
      • SimformerParams.dim_id
      • SimformerParams.dim_joint
      • SimformerParams.dim_value
      • SimformerParams.dropout_rate
      • SimformerParams.fourier_features
      • SimformerParams.num_heads
      • SimformerParams.num_hidden_layers
      • SimformerParams.num_layers
      • SimformerParams.qkv_features
      • SimformerParams.rngs
      • SimformerParams.widening_factor
• gensbi.utils package
  • Submodules
  • gensbi.utils.math module
    • divergence()
  • gensbi.utils.model_wrapping module
    • GuidedModelWrapper
      • GuidedModelWrapper.cfg_scale
      • GuidedModelWrapper.get_vector_field()
    • ModelWrapper
      • ModelWrapper.get_divergence()
      • ModelWrapper.get_vector_field()
  • gensbi.utils.plotting module
    • plot_trajectories()
  • Module contents

Module contents