Python environment

Recommended environment:

uv venv --python 3.12 .venv
source .venv/bin/activate
uv pip install \
  numpy pandas scipy matplotlib seaborn statsmodels scikit-learn \
  cmdstanpy arviz jax jaxlib blackjax oryx tensorflow-probability

CmdStan is already available on this machine at ~/.cmdstan/cmdstan-2.34.1; if not:

import cmdstanpy
cmdstanpy.install_cmdstan()

Project layout: