The internal version also has additional examples in code that is more sensitive (e.g. changes to the TPU compiler backend) that we aren't able to include in the external version.

→ View original post on X — @jeffdean