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[cuda13]
# or
uv pip install gensbi[cuda13]

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[cuda13]

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[cuda13,examples]
# or
pip install gensbi[cuda13,examples]

Requirements

• Python 3.12+

• JAX >=0.11.1, <0.13 (0.11.x and 0.12.x are both tested)

• Flax >=0.12.10

• (See pyproject.toml for full requirements)