convolution based PyTorch implementation for bleu sft.

Input

4825-dim embedding

Encoder

73 x Attention-UNet with 36 heads

Output

recall projection