Imported from https://github.com/google-research/language/blob/master/language/labs/consistent_zero_shot_nmt