https://congchan.github.io/posts/multi-token-prediction/