Codebase accompanying the AISTATS 2022 paper : "Predicting the impact of treatments over time with uncertainty aware neural differential equations" - View it on GitHub