Example code for Discrete Flows?

Would it be possible to add example code in for Discrete Flows like the demo examples from the paper? Eg. the discretized mixture of gaussians?

I am finding that the brief example listed here: https://github.com/google/edward2/blob/4270880356e720850f96fb4012510a315546fc54/edward2/tensorflow/layers/README.md

has a number of bugs. eg. it has the command `flow.reverse(features)` but this throws an error because `flow` is a Keras Sequential object rather than a single layer.

Thank you for any help that can be provided!