Staff Software Engineer, AI Data Generation Platform

Google
Sunnyvale, CA, USA

About the job

Our organization accelerates the path to AI by engineering robust data-acquisition systems that deliver high-fidelity datasets including human-generated data, evaluations, synthetic data at scale, and data to fill critical voids in areas like agentic workflows, coding, and other specialized domains. Our organization builds systems to secure such high-quality data and also improve velocity for ML researchers and product developers to use the data efficiently for model training/fine-tuning and product adoption. In this role, you will make direct contributions to infrastructure and model customization. Your work will also focus on emerging areas like reinforcement learning techniques for building infrastructure for agentic workflows.

Responsibilities

Design and build scalable, general-purpose data tooling and infrastructure to support variety of applied use-cases for building agents.

Learn novel Generative AI use cases to learn in depth their data needs and identify solutions leveraging the AI Data Platform.

Build and optimize infrastructure for agentic environment data creation addressing unique and emerging technical tests of modeling techniques involving reinforcement learning.

Work cross functionally with Data Science, Product, other infrastructure teams.

Qualifications

Minimum

Bachelor's degree or equivalent practical experience.

8 years of experience in software development.

5 years of experience testing, and launching software products, and 3 years of experience with software design and architecture.

5 years of experience leading ML design and optimizing ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning).

2 years of experience with state of the art GenAI techniques (e.g., LLMs, Multi-Modal, Large Vision Models) or with GenAI-related concepts (language modeling, computer vision).

Preferred

Master’s degree or PhD in Engineering, Computer Science, or a related technical field.

8 years of experience with data structures/algorithms.

3 years of experience in a technical leadership role leading project teams and setting technical direction.

3 years of experience working in a complex, matrixed organization involving cross-functional, or cross-business projects.