Per-example attention + logit lens on CODI-GPT2 latent tokens (GSM8K). Click to expand. Hover a latent row to see what it attends to. Click predictions for top-5.