gensbi

GenSBI: A library for Simulation-Based Inference (SBI) using Optimal Transport Flow Matching and Diffusion models in JAX.

Provides tools for probabilistic modeling, simulation, and training of generative models, including:

• Flow Matching techniques

• Diffusion models (EDM, score matching)

• Transformer-based models (Flux1, Simformer)

See the documentation for details and usage examples.

Submodules

• gensbi.diffusion
• gensbi.flow_matching
• gensbi.models
• gensbi.recipes
• gensbi.utils

Attributes

__version__

Package Contents

gensbi.__version__ = '0.0.1'