I am looking for a pytorch implementation of an RNN module with variational dropout (= SAME dropout mask at each timestep AND recurrent layers) as proposed by Gal and Ghahramani in the paper A Theoretically Grounded Application of Dropout in Recurrent Neural Networks https://papers.nips.cc/paper/6241-a-theoretically-grounded-application-of-dropout-in-recurrent-neural-networks.pdf 22 .