We are still working on code cleaning of our own codebase, along with hint-nart.
Jiatao Gu has provided an excellent re-implementation of our model in the fairseq repo. We highly recommend you to try it!
@incollection{NIPS2019_8566,
title = {Fast Structured Decoding for Sequence Models},
author = {Sun, Zhiqing and Li, Zhuohan and Wang, Haoqing and He, Di and Lin, Zi and Deng, Zhihong},
booktitle = {Advances in Neural Information Processing Systems 32},
editor = {H. Wallach and H. Larochelle and A. Beygelzimer and F. d\textquotesingle Alch\'{e}-Buc and E. Fox and R. Garnett},
pages = {3011--3020},
year = {2019},
publisher = {Curran Associates, Inc.},
url = {http://papers.nips.cc/paper/8566-fast-structured-decoding-for-sequence-models.pdf}
}