The Staff AI/ML Research Engineer role at Google focuses on developing advanced AI agents and optimizing machine learning infrastructure. The position is part of the Cloud AI Research team, which aims to bridge the gap between foundational research and production systems. Candidates will work on complex strategic requests and architect robust agent state machines. This role requires a blend of technical expertise and leadership to drive innovation in AI technologies.

• 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.
• Experience developing autonomous agents, multi-agent workflows, or agentic frameworks.
• Experience designing, pre-training, fine-tuning, or evaluating Large Language Models and multimodal foundation models.

• Lead 2–4 week 'Strike Sprints' to rapidly prototype advanced agent state machines.
• Partner with product and engineering teams to translate novel agentic prototypes into scalable enterprise solutions.
• Build generalizable horizontal agents to accelerate engineering and researcher velocity.
• Develop state-of-the-art agent architectures and self-improving foundation models.
• Collaborate across Google to disseminate technical playbooks and optionally publish breakthroughs at ML conferences.

• Employees at Google are often offered benefits like comprehensive health insurance, 401(k) matching, and flexible work arrangements, among other benefits.