This is a JAX-converted version of [T5-small](https://huggingface.co/google-t5/t5-small).