{bc}
Apply
company_site

Staff AI/ML Engineer, Cloud AI Research

Google
Sunnyvale, USA
Senior · 8+ years experience
Onsite
USD 207000-300000 / year
Discovered 5 days ago
javajavascriptpython

Full Job Posting

Responsibilities

  • 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.

Minimum qualifications:

  • Bachelor's degree in Computer Science, Artificial Intelligence, Machine Learning, or equivalent practical experience.
  • 8 years of experience in software development, including coding experience in Python, C++, Java, JavaScript, or R.
  • 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).

Apply for this job in 1 click

Skip the repetitive application forms

Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.

Sarah M.James T.Maya R.

Trusted by over 500,000 job seekers on Base Career

Start Free Today

More from this employer

More jobs at Google

Senior Program Manager I, Google Cloud

Sunnyvale, USA

Discovered Today

Policy Development Lead

San Bruno, USA

Discovered Today

Software Engineer III, Infrastructure, Google Cloud NetInfra

Madison, USA

Discovered Today

People Partner, Core AI Foundations

San Jose, USA

Discovered Today

Senior Software Engineer, Infrastructure, Google Cloud Security and Privacy

Kirkland, USA

Discovered Today

Senior Staff Software Engineer, Knowledge Catalog For AI

Sunnyvale, USA

Discovered Today

Senior Business Data Scientist, Trust and Safety Business Analytics

Austin, USA

Discovered Today

Customer Engineer, Outcome, Google Cloud

Bengaluru, IND

Discovered Today