Stochastic Block Model in edward2

Hello!

I am a student and currently doing a research internship at the Technical University of Munich and trying to implement a Stochastic Block Model (SBM). I tried to understand the various functions and classes using the karate club toy example from the edward1 publication. However, I prefer to use edward2 and thus have to port the code a little bit. Also, I have seen that some other guys have opened issues for the edward1 release detailing improvements to the stock example provided in edward1, especially these ones:

[Stochastic Block Model #57](https://github.com/blei-lab/edward/issues/57)
[Implement a stochastic block model example #715](https://github.com/blei-lab/edward/pull/715)
[Added Stochastic Block Model Example (Jupyter Notebook) #308](https://github.com/tensorflow/probability/pull/308)

So I was wondering whether there might be an edward2 compatible and updated version of the SBM realization, in particular with some more documentation/comments. That would be really helpful, especially as I'm quite new to edward and some tensorflow functions. I really think, that the framework provides many powerful stochastic concepts.

Thank you!