Staff Research Engineer, Data Agents

Job Overview

We are seeking a Staff Research Engineer to join the Data Agent team within the Databricks AI Research organisation. This team is focused on developing frontier enterprise data agents capable of autonomous planning, code generation, and multi-step workflow execution.

Responsibilities

• Develop post-training recipes for enterprise Data agents that enable autonomous planning, code generation, and multi-step workflow execution in complex enterprise settings.
• Collaborate with product teams to transform prototypes and research concepts into optimal agentic experiences for Databricks users.
• Design systems that allow agents to discover and utilise relevant lakehouse context, such as tables, notebooks, code, and cell outputs, to generate more accurate and useful results.
• Elevate the team's technical capabilities through strong design, execution, debugging, and mentorship, contributing to the long-term direction of agentic experiences at Databricks.

Requirements

• A BS, MS, or PhD in Computer Science or a related field.
• At least 2 years of experience in an applied research environment, with a proven track record of transitioning research prototypes into production.
• Experience with Large Language Models (LLMs), agents, reinforcement learning, and post-training workflows.
• Ability to work effectively in a fast-paced environment that combines research exploration with product and engineering rigour.
• Strong communication and collaboration skills, particularly in cross-functional teams with researchers, engineers, and product stakeholders.

Compensation

• The pay range for this role is $190,000 - $270,000 USD.

Benefits

• Comprehensive benefits and perks tailored to employee needs (specific details vary by region).