This repository stores the collection of scripts and modules tied to the in-progress revision of the preprint, "A framework for variational inference and data assimilation of soil biogeochemical models using state space approximations and normalizing flows".
Neural moving average flow Python experiment scripts and dependent modules located in "pytorch_sbm_sde_vi" folder. Experiment scripts correspond to the various "learn... .py" files. To replicate an individual experiment, such as a fixed-$\theta$ or or joint hamiltorch
SSM-NUTS benchmarking is labeled with "mcmc_..." in the filename.