References:
https://arxiv.org/abs/1810.04719
https://github.com/google/uis-rnn
https://catalog.ldc.upenn.edu/LDC2001S97