disco: a toolkit for Distributional Control of Generative Models