This repository contains code for Bayesian modeling of COVID-19 using numpyro and jax.
The team is based at the College of Information and Computer Sciences and School of Public Health & Health Sciences at UMass. The model creators, contributors, and weekly crank-turners are:
Dr. Reich and Dr. Ray direct the CDC Influenza Forecasting Center of Excellence at UMass.
Our code depends on numpyro and jax. If you don't have these packages, our installation routine will pull and install them:
git clone https://github.com/dsheldon/mechbayes
cd mechbayes
pip install -e .
If you need to manually install jax and numpyro, please see the instructions for those packages: jax, numpyro.