Code for "Sequential Neural Models with Stochastic Layers" - View it on GitHub