Get started with Gemma on Ray on Vertex AI- Google Developers Blog
Jul 9, 2024Ju-yeong Ji, Ivan Nardini

Summary
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.
More from the Google Developer Platform trail
- Autonomous LLM post-training with Tunix on TPUs- Google Developers Blog
- HeyGen x Google Cloud: Bringing Avatar IV to TPUs- Google Developers Blog
- Run Ray on TPU, Part 2: Ray AI libraries- Google Developers Blog
- Scaling Agentic RL: High-Throughput Agentic Training with Tunix- Google Developers Blog
- Run Ray on TPU, Part 1: The foundations- Google Developers Blog