google/gemma_pytorch

Health v2 · maintenance only

Flags: no_releases archived

Adoption not part of the score

5719 stars · 602 forks observed · 2026-08-28

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

The official PyTorch implementation of Google's Gemma family of open large language models, including text-only and multimodal variants. It provides model and inference code supporting CPU, GPU, and TPU via PyTorch and PyTorch/XLA.

Use cases

• run gemma models locally with pytorch
• inference for gemma 3 on gpu
• run gemma on tpu with pytorch xla
• load gemma checkpoints from kaggle or hugging face
• run multimodal gemma 4b 12b 27b inference
• fine-tune or experiment with gemma model architecture

When to choose

• you want the official reference implementation of Gemma in PyTorch
• you need inference across CPU, GPU, and TPU including PyTorch/XLA
• you want to run Gemma 1/2/3 or CodeGemma checkpoints directly
• you need multimodal Gemma 3 support in PyTorch

When to avoid

• you want a high-performance production serving stack with batching and quantization
• you prefer using Gemma through Hugging Face Transformers or vLLM
• you need training/fine-tuning pipelines, which this repo does not focus on
• you use a framework other than PyTorch such as JAX or TensorFlow

Facets

library · maturity active

llm-inference machine-learning deep-learning large-language-models deep-learning artificial-intelligence python cloud gemma pytorch google tpu open-weights inference multimodal gpu linux

Member repositories

For agents

markdown · JSON · MCP: product_card(name="google/gemma_pytorch")