We're announcing the public release of Distributions 2.0 at AISTATS. Let's get all our existing models in the 2.0 branch by then.