BAO and SNe Ia likelihoods for JAX.
Remember on cuda to install the appropriate jax:
pip install "jax[cuda]"
pip install -e .
# or
pip install -e ".[cuda]"
# or using uv
uv sync --extra cudaCall run.py followed by the model and datasets. Optionally specify nlive (default is 1000) and any additional parameters for the model.
For example, flexknot requires the number of knots n:
python run.py lcdm des5y
python run.py cpl desidr2 pantheonplus --nlive=500
python run.py flexknot desidr2 --n=10