Google Developer Platform
Get started with Gemma on Ray on Vertex AI- Google Developers Blog

Get started with Gemma on Ray on Vertex AI- Google Developers Blog

7/9/2024 · Ju-yeong Ji, Ivan Nardini

What this post added

This post introduces the integration of Gemma, a family of open models, with Ray on Vertex AI for supervised fine-tuning and serving. It details the process of setting up a Ray cluster on Vertex AI using custom Docker images, preparing datasets (XSum), configuring cloud resources (Cloud Storage, Artifact Registry, Vertex AI TensorBoard), and submitting fine-tuning jobs using the Ray Jobs API. The post highlights the use of Ray Train for distributed HuggingFace Transformers training with PyTorch and demonstrates how to monitor training artifacts and progress.

Read the original post ↗