google-research/t5x

Health v2 · maintenance only

Flags: no_releases

Adoption not part of the score

2998 stars · 338 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

T5X is a modular, composable framework built on JAX and Flax for high-performance training, evaluation, and inference of sequence models at many scales. It is a modernized reimplementation of the original T5 codebase, designed for research-friendly, configurable self-service experiments on TPUs and GPUs.

Use cases

• train sequence-to-sequence models on TPUs
• fine-tune T5-style language models at scale
• run model evaluation and inference for large language models
• run multi-host distributed training experiments on Google Cloud
• reproduce T5 research experiments with a modern JAX stack

When to choose

• you need to train or fine-tune large sequence models on TPU/GPU hardware
• you want a configurable, research-oriented alternative to the original Mesh TensorFlow T5 codebase
• your workflow already uses JAX and Flax

When to avoid

• you need a simple PyTorch or TensorFlow training loop for small models
• you want a general-purpose serving/inference server rather than a training framework
• you are not working with sequence models or language tasks

Facets

framework · maturity maintenance

llm-training machine-learning deep-learning benchmarking deep-learning large-language-models machine-learning python cloud jax flax tpu sequence-models gin-config google-research model-training evaluation inference natural-language-processing gpu linux

Member repositories

For agents

markdown · JSON · MCP: product_card(name="google-research/t5x")