openghg / openghg_inversions

University of Bristol Atmospheric Chemistry Research Group RHIME Inversion code (with openghg dependency)
MIT License
5 stars 0 forks source link

Add nuts sampler options, multiple object stores in ini file #29

Closed brendan-m-murphy closed 5 months ago

brendan-m-murphy commented 6 months ago

This branch extends the min_model_error branch to

1) allow the kwarg "nuts_sampler": "numpyro", which allows faster sampling if numpyro and jax are installed in your venv.

2) allow a list of object stores in ini files for footprints and obs data, e.g. obs_store = ["store1", "store2"]

3) skip sites if no data is found, without causing the entire inversion to fail. The dropped sites can be found in the .out file from SLURM. Site indicator numbers are based on the list of sites in the ini file, so they'll be consistent across runs.