Imported from https://github.com/cindyxinyiwang/deep-latent-sequence-model