Pyro for Iterative Proportional Fitting/Matrix Scaling

I’m working on a matrix scaling/iterative proportional fitting problem, and had initially thought Pyro would be a helpful framework for modeling my problem. Another way to phrase my problem is construction of a joint distribution based on marginal distributions.

I haven’t found much documentation suggesting there to be methods in Pyro, but I am curious if others have used Pyro towards such ends.