T5X Retrieval 是 T5(Text-to-Text Transfer Transformer)的 JAX 实现,针对检索应用进行了优化。它通过 Seqio 支持灵活的数据加载,并通过 T5X(https://github.com/google-research/t5x)实现快速/可扩展的训练。欢迎试用!