Installation

Using uv (recommended)

To add gensbi as a dependency to an existing pyproject.toml:

uv add gensbi

For a standalone install without a project:

uv pip install gensbi

For GPU support:

uv add gensbi[cuda12]
# or
uv pip install gensbi[cuda12]

Tip

To install uv, run:

curl -LsSf https://astral.sh/uv/install.sh | sh

Using pip

pip install gensbi

For GPU support:

pip install gensbi[cuda12]

If you want to run the example notebooks, install the examples extra, which pulls in sbibm-jax[loader] (the sbibm_jax package providing the SBIBM task loaders and datasets the notebooks use):

uv add gensbi[examples]
# or
pip install gensbi[examples]

To install all the optional dependencies at once, run:

uv add gensbi[cuda12,examples]
# or
pip install gensbi[cuda12,examples]

Requirements

• Python 3.12+

• JAX

• Flax

• (See pyproject.toml for full requirements)