Install setu

Install from PyPI. The core install covers training and the full validate_estimator suite; the PPL bridges and tuning are opt-in extras.

pip install setu-sbi                     # core (JAX + FlowJAX)
pip install "setu-sbi[pymc]"             # + PyMC / BlackJAX
pip install "setu-sbi[numpyro]"          # + NumPyro
pip install "setu-sbi[tuning]"           # + Optuna hyperparameter tuning
pip install "setu-sbi[all]"              # everything above

Each extra pulls one backend: pymc pulls PyMC and BlackJAX, numpyro pulls NumPyro. The tuning extra is needed only for tune_nle / tune_nre / tune_mixed_nle. all is the union of every extra.

Distribution name vs import name

The distribution on PyPI is setu-sbi; the import name is setu.

import setu

Install setu-sbi, import setu.