768-dimensional hidden vector
              ↓
50,257 logits

        __
        __