"""Multi-token prediction."""
# TODO