[FlaxT5ForConditionalGeneration] is also supported by this notebook.