An extension of PRIME for the scalable calibration of high-dimensional disease outbreak models using Variational Inference - View it on GitHub