Update Figure numbers and descriptions to correspond to current version of arxiv paper.
Update Figure numbers and descriptions to correspond to current version of arxiv paper.

PiperOrigin-RevId: 409179633

google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research