# Staff Research Engineer, Data Agents

**Company:** [Databricks](https://hotfix.jobs/companies/databricks)
**Location:** San Francisco, CA
**Role:** AI Research
**Salary:** $190k – $270k/yr
**Experience:** 2+ years
**Skills:** LLMs, Reinforcement Learning, Post-Training, Python, Machine Learning, AI Agents, Code Generation, Multi-Agent Systems
**Posted:** 2026-06-27

> Develop post-training recipes and build enterprise data agents for autonomous planning, code generation, and multi-step workflows. Requires 2+ years applied research experience shipping prototypes, plus expertise in LLMs, agents, and RL.

## Job Description

## Responsibilities
- Develop the best post-training recipes to train enterprise Data agents, that are capable of autonomous planning, code generation, and multi-step workflow execution within intricate enterprise settings.
- Partner closely with product teams to turn prototypes and research ideas into the best agentic experience for Databricks users.
- Build systems that help the agent discover and use relevant lakehouse context, including tables, notebooks, code, and cell outputs, to produce more accurate and useful results.
- Raise the technical bar for the team through strong design, execution, debugging, and mentorship, helping shape the long-term direction of agentic experiences at Databricks.

## Requirements
- BS, MS, or PhD in Computer Science or a related field.
- 2+ years of experience in an applied research environment, with a track record of shipping research prototypes to production.
- Experience in LLMs, agents, reinforcement learning, post-training workflows.
- Ability to work effectively in a fast-moving environment that blends research exploration with product and engineering rigor.
- Clear communication and strong cross-functional collaboration with researchers, engineers, and product stakeholders.

## Similar roles

- [Staff ML Research Scientist](https://hotfix.jobs/jobs/1449ecbe-a090-4b13-8109-5b529aae9189) - Rad AI - San Francisco, CA - $190k – $260k/yr
- [Senior/Staff Machine Learning Research Scientist: Generative Modeling for Planning](https://hotfix.jobs/jobs/d7957acf-fcb1-46b9-8cfb-82aa9aaf664a) - Nuro - Mountain View, CA - $194k – $352k/yr
- [Senior Staff Machine Learning Scientist, Assets](https://hotfix.jobs/jobs/64b03df2-0c19-48f2-8ba6-eabed0a170f6) - Webflow - Remote - $194k – $285k/yr
- [Member of Technical Staff - Research](https://hotfix.jobs/jobs/ea4a552a-3a64-4bc8-a541-0f50869befb5) - Polymath - San Francisco, CA - $200k – $350k/yr
- [Staff NLP Scientist](https://hotfix.jobs/jobs/7620a7de-b405-4882-913b-419f4d94846c) - Machinify - Remote - $180k – $280k/yr

**Apply:** https://hotfix.jobs/jobs/4c56f144-60a7-4c49-a56e-e06c7f743d32
**Canonical:** https://hotfix.jobs/jobs/4c56f144-60a7-4c49-a56e-e06c7f743d32