NMT Assignment Note: Heavily inspired by the https://github.com/pcyin/pytorch_nmt repository cs224n-nmt-tmp