Hi! Based on my own understanding, there should be a noise input to each of the LSTM cells, but the graph in README shows that there is only one random vector z input to the first LSTM cell. I am not familiar with TensorFlow, but it looks like your code is also injecting randomness to each of the cells.