Type hints and decorator for TF T5
@Rocketknight1 @gante 
Thanks for this, the PR looks perfect to me! I see XLA generation tests failing, though - @gante could this be another situation like GPT-2 with some weird input names that are exposed by the decorator?<|||||>_The documentation is not available anymore as the PR was closed or merged._<|||||>@gante I think some of those were changes to main that came in during the PR, including the missing XLA function that caused those test failures! Let me fix it up real quick.<|||||>@Dahlbomii Tests are green so we're going to merge now. Thanks for your help with this, and thanks for your patience as we both edited the file you were working on and renamed `master` to `main` midway through your PR!