Staff AI/ML Engineer, Cloud AI Research

Google LLC; Sunnyvale (CA); On-site; USD 207,000 - 300,000; Full time

Job summary

Google LLC seeks a senior ML/AI engineer to lead design and optimization of distributed ML systems and autonomous agent frameworks. You will shape agent state machines, tool usage, and scalable deployments for enterprise use.

The role spans research-to-production with rapid prototyping and collaboration across teams. Responsibilities include building robust agent architectures, evaluating LLMs and multimodal models, and contributing to publication-ready work.

Qualifications

• Experience leading ML system design and optimizing ML infrastructure.
• Experience designing, pre-training, fine-tuning or evaluating LLMs and multimodal models.
• Experience developing autonomous agents and multi-agent workflows.

Responsibilities

• Lead 2–4 week Strike Sprints to prototype agent state machines and tool-use orchestration.
• Partner with product and engineering teams via Forward Deployed Engineering to scale prototypes.
• Build horizontal agents (e.g., AutoResearch) to accelerate velocity and automate value.
• Develop agent architectures and self-improving foundation models for autonomous systems.

Skills

Distributed training

Model deployment

Autonomous agents

LLMs & multimodal models

Multi-agent workflows

Education

Bachelor's degree in Computer Science / AI / ML

Master's degree or PhD in a related technical discipline

Tools

AutoGen

CrewAI

LangGraph

ADK

tool-use / function-calling

multi-agent R

Job description

corporate_fare Google place Sunnyvale, CA, USA

• Bachelor's degree in Computer Science, Artificial Intelligence, Machine Learning, or equivalent practical experience.
• 5 years of experience leading ML system design and optimizing ML infrastructure (e.g., distributed training, model deployment, evaluation, data processing).
• Experience developing autonomous agents, multi-agent workflows, or agentic frameworks (e.g., AutoGen, CrewAI, LangGraph, ADK, tool-use/function-calling, or multi-agent R.
• Experience designing, pre-training, fine-tuning, or evaluating Large Language Models (LLMs) and multimodal foundation models.

Preferred qualifications:

• Master's degree or PhD in Computer Science, Artificial Intelligence, Machine Learning, or a related technical discipline.
• Experience developing test-time scaling methods, search-guided decoding, and safe AI systems (hallucination mitigation, alignment guardrails).
• Experience building evaluation benchmarks, test harnesses, or simulation environments for complex reasoning, code generation, or multimodal tasks.
• Experience in RL, reward modeling, and human/AI preference alignment techniques (e.g., RLHF, RLAIF, DPO).
• Proven experience operating as a Staff technical lead in rapid SWAT, Pathfinding, or Forward Deployed Engineering (FDE) environments writing robust production code.
• Two or more first-author scientific publication submission(s) on agents in machine learning and natural language processing for conferences, journals, or public repositories (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP).

About the job

Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast‑paced business grow and evolve. We need our engineers to be versatile, display leadership qualities and be enthusiastic to take on new problems across the full‑stack as we continue to push technology forward.

About the job

Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast‑paced business grow and evolve. We need our engineers to be versatile, display leadership qualities and be enthusiastic to take on new problems across the full‑stack as we continue to push technology forward.

The frontier of AI is moving faster than the traditional enterprise deployment life-cycle can tolerate. Breakthroughs frequently sit stranded in academic papers while production systems struggle with brittle orchestration and scaling bottlenecks. Our organization exists to eliminate that "GenAI Engineering Gap." Operating as an AI‑Native pathfinding engine at the intersection of foundational research, optimized ML infrastructure and rapid prototyping, we don't simply publish papers; we build the future of AI agents and deploy them at breakneck speed. In this role, you will operate as an "Ambiguity Buster," deconstruct complex requests from executive leadership, architect agent state machines and prove viability through functional Thinnest Viable Proofs (TVPs). You will write production‑grade code to build both generalizable horizontal agents and vertical enterprise solutions, taking novel agentic prototypes directly into the field to solve real customer bottlenecks. Google Cloud accelerates every organization’s ability to digitally transform its business and industry. We deliver enterprise‑grade solutions that leverage Google’s technology, and tools that help developers build more sustainably. Customers in more than 200 countries and territories turn to Google Cloud as their trusted partner to enable growth and solve their most critical business problems. Individual pay is determined by factors including job‑related skills, experience, and relevant education or training.

US: $207000 - $300000 (USD) + 20% bonus target + equity + benefits

Learn more about benefits at Google .

• Lead 2–4 week "Strike Sprints" to rapidly prototype advanced agent state machines, tool‑use orchestration and reflection loops, transforming boilerplate programming into intent‑driven AI topologies.
• Partner with product and engineering teams via Forward Deployed Engineering (FDE), embedding into their workflows to translate novel agentic prototypes into scalable enterprise solutions targeting massive operational efficiencies.
• Build generalizable horizontal agents (e.g., AutoResearch and AutoFounder) to accelerate engineering/researcher velocity and automate end‑to‑end business value generation across Google.
• Develop state‑of‑the‑art agent architectures, planning frameworks and self‑improving foundation models to advance autonomous AI systems.
• Collaborate across Google to disseminate technical playbooks and agentic frameworks and optionally publish breakthroughs at ML conferences.

Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law. See also Google's EEO Policy , Know your rights: workplace discrimination is illegal , Belonging at Google , and How we hire .

Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles unless stated otherwise in the job posting.

To all recruitment agencies: Google does not accept agency resumes. Please do not forward resumes to our jobs alias, Google employees, or any other organization location. Google is not responsible for any fees related to unsolicited resumes.

Equity is granted exclusively and discretionarily by Alphabet Inc. on the basis of an agreement concluded between you and Alphabet Inc. Alphabet Inc. is your sole contractual partner with respect to equity grants. GSU grants are not guaranteed, are discretionary, are subject to approval by the Alphabet Inc. board of directors or its delegate, the terms of the relevant Alphabet Inc. stock plan, and your grant agreement. They have no impact on statutory payments. Current or past grants do not confer an acquired right.