Want to contribute by re-writing it in Pytorch

[shashankg7]

Hi,

I am interested in learning and implementing variational RNNs.

I can take up the task of re-implementing it in Pytorch. I would need some guidance since I am not very familiar with the topic.

Please let me know