1. Move benchmarks below advanced usage. 2. Request A100 GPU 3. Minor cosmetic changes. PiperOrigin-RevId: 729538515 Change-Id: I12709cd95f6865e53fd3932ea0b303ee8efc4028
PiperOrigin-RevId: 728706864 Change-Id: I8f3a507234c4265564a4fee9df0ab8111ff7dd85