The use of smaller proxy tasks was covered in Section 3.3 of So et al., so no Google inside knowledge needed (although it's pretty subtly described in So et al.) There's some discussion in Appendix C of

→ View original post on X — @jeffdean,