AI models
TxGemma 2B
Training compute
3.2×10²² FLOP
Parameters
2.6B
Published
Apr 8, 2025
TxGemma 2B is an AI model developed by Google DeepMind and Google Research (United States), first published in April 2025. It works in the language and biology domain, on tasks such as protein or nucleotide language model (plm/nlm), protein property prediction, small molecule property prediction and 4 more.
Training it took an estimated 3.2×10²² FLOP of compute (estimation method: operation counting). The model has 2,600,000,000 parameters. Training ran on 256 Google TPU v4.
Access: Open weights (restricted use). Its weights are openly available. It is built on top of Gemma 2 2B. Epoch AI rates the confidence of this record as confident.
Full record
- Organization
- Google DeepMind, Google Research
- Country of organization
- United States
- Domain
- Language, Biology
- Task
- Protein or nucleotide language model (pLM/nLM), Protein property prediction, Small molecule property prediction, Question answering, Protein question answering, Protein function prediction, Language modeling/generation
- Training compute
- 3.2×10²² FLOP
- Compute estimation method
- Operation counting
- Parameters
- 2,600,000,000
- Training hardware
- Google TPU v4
- Chips used
- 256
- Training power draw
- 170.7 kW
- Model accessibility
- Open weights (restricted use)
- Open weights
- Yes
- Base model
- Gemma 2 2B
- Epoch confidence
- Confident
More from Google DeepMind,Google Research
SourceEpoch AI, 'AI Models'. Published online at epoch.ai. Retrieved 2026-07-29 from https://epoch.ai/data/ai-models. Licensed under CC BY 4.0.
← All ai models