Contextual embeddings for translation