Put class into random module prior

[jpchen]

random_module takes pyro distributions, so you would have to implement your flow as a torch distribution object. take a look at these flows for an example.