Here we provide a cleaned-up version of our codebase used for our experiments. Each model is a self-contained file with its own main method.