Try Gemini 1.5 Pro, our most advanced multimodal model in Vertex AI, and see what you can build with a 1M token context window. Try Gemini 1.5 Pro, our most advanced multimodal model in Vertex AI, and see what you can build with a 1M token context window.

Use Gemma open models

Gemma is a set of lightweight, generative artificial intelligence (AI) open models. Gemma models are available to run in your applications and on your hardware, mobile devices, or hosted services. You can also customize these models using tuning techniques so that they excel at performing tasks that matter to you and your users. Gemma models are based on Gemini models and are intended for the AI development community to extend and take further.

Fine-tuning can help improve a model's performance in specific tasks. Because models in the Gemma model family are open weight, you can tune any of them using the AI framework of your choice and the Vertex AI SDK. You can open a notebook example to fine-tune the Gemma model using a link available on the Gemma model card in Model Garden.

The following Gemma models are available to use with Vertex AI. To learn more about and test the Gemma models, see their Model Garden model cards.

Model name	Use cases	Model Garden model card
Gemma	Best for text generation, summarization, and extraction.	Go to the Gemma model card
CodeGemma	Best for code generation and completion.	Go to the CodeGemma model card

The following are some options for where you can use Gemma:

Use Gemma with Vertex AI

Vertex AI offers a managed platform for rapidly building and scaling machine learning projects without needing in-house MLOps expertise. You can use Vertex AI as the downstream application that serves the Gemma models. For example, you might port weights from the Keras implementation of Gemma. Next, you can use Vertex AI to serve that version of Gemma to get predictions. We recommend using Vertex AI if you want end-to-end MLOps capabilities, value-added ML features, and a serverless experience for streamlined development.

To get started with Gemma, see the following notebooks:

Use Gemma in other Google Cloud products

You can use Gemma with other Google Cloud products, such as Google Kubernetes Engine and Dataflow.

Use Gemma with GKE

Google Kubernetes Engine (GKE) is the Google Cloud solution for managed Kubernetes that provides scalability, security, resilience, and cost effectiveness. We recommend this option if you have existing Kubernetes investments, your organization has in-house MLOps expertise, or if you need granular control over complex AI/ML workloads with unique security, data pipeline, and resource management requirements. To learn more, see the following tutorials in the GKE documentation:

Use Gemma with Dataflow

You can use Gemma models with Dataflow for sentiment analysis. Use Dataflow to run inference pipelines that use the Gemma models. To learn more, see Run inference pipelines with Gemma open models.

Use Gemma with Colab

You can use Gemma with Colaboratory to create your Gemma solution. In Colab, you can use Gemma with framework options such as PyTorch and JAX. To learn more, see:

Gemma model sizes and capabilities

Gemma models are available in several sizes so you can build generative AI solutions based on your available computing resources, the capabilities you need, and where you want to run them. Each model is available in a tuned and an untuned version:

Pretrained - This version of the model wasn't trained on any specific tasks or instructions beyond the Gemma core data training set. We don't recommend using this model without performing some tuning.
Instruction tuned - This version of the model was trained with human language interactions so that it can participate in a conversation, similar to a simple chat bot.

If you need to decide between Gemma 2B and Gemma 7B, consider Gemma 2B. The lower parameter sizes of Gemma 2B mean it has lower resource requirements and more deployment flexibility than Gemma 7B.

Model name	Parameters size	Input	Output	Tuned versions	Intended platforms
Gemma 2B	2.2 billion	Text	Text	Pretrained Instruction tuned	Mobile devices and laptops
Gemma 7B	7 billion	Text	Text	Pretrained Instruction tuned	Desktop computers and small servers
CodeGemma 2B	2 billion	Text	Text	Pretrained	Desktop computers and small servers
CodeGemma 7B	7 billion	Text	Text	Pretrained Instruction tuned	Desktop computers and small servers

Gemma has been tested using Google's purpose built v5e TPU hardware and NVIDIA's L4(G2 standard), A100(A2 standard), H100(A3 standard) GPU hardware.