Count what token follows each token. This is the complete sufficient statistic for a tiny bigram language model before probabilities or sampling.

def bigram_counts(tokens):
    return {token: {next_token: count}}

• look only at adjacent pairs
• create a row for each source token
• increment repeated transitions