This is the accompanying repo to the paper "Many processors, little time: MCMC for partitions via optimal transport couplings", to appear in AISTATS 2022.
To install necessary Python packages, make the environment specified by the environment.yml.
Generate figures in main text
Run new experiments
The list of examples include
- Mixture model on gene dataset
- Gibbs sampler
- Jain-Neal split merge
- Graph coloring on k-regular graph
Please contact tdn@mit.edu if you have any questions!