Update prepare_glue_submission with tasks and to handle the SuperGLUE AX-b and AX-g diagnostic tasks.
Update prepare_glue_submission with tasks and to handle the SuperGLUE AX-b and AX-g diagnostic tasks.

PiperOrigin-RevId: 404609037

google-research/text-to-text-transfer-transformer