The SEQ2SEQ model 1 https://github.com/lijierui/ group-attention takes the text of the whole problem as the input and corresponding equation as the output.
the group attention consists of four different types of multi-head attention modules.