This is the code repository for this paper.
Models are implemented in Python and JAX with dependencies on libraries listed below at the end.
See data/ for data preprocessing, synthetic data generation, and visualization.
See notebooks in notebooks/ for a examples on loading and analyzing fitted models.
See scripts/ for model fitting, analysis and plotting code.
Formatting done with ufmt, run ufmt format .