Test implementation of "Aligned Cross Entropy for Non-Autoregressive Machine Translation" https://arxiv.org/abs/2004.01655
axe/axe.py
Please see detail at my colaboratory notebook.
https://colab.research.google.com/github/m3yrin/aligned-cross-entropy/blob/master/demo.ipynb