Skip to content
Databricks

Staff Research Engineer, Data Agents

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.

190k – 270kSan Francisco, CAAI ResearchOnsite2+ YOE
Apply

About the role

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.

Skills

LLMsReinforcement LearningPost-TrainingPythonMachine LearningAI AgentsCode GenerationMulti-Agent Systems

Similar roles

AI Research jobs
Rad AI

Staff ML Research Scientist

Leads end-to-end applied ML research in NLP, LLMs, retrieval, and multimodal models for healthcare AI, driving from experimentation to production deployment with rigorous evaluation and clinician collaboration. Requires 7+ years experience, MS/PhD, and depth in ML areas like PyTorch tooling.

190k – 260kSan Francisco, CAAI ResearchOn-site7+ YOENLPRay
Nuro

Senior/Staff Machine Learning Research Scientist: Generative Modeling for Planning

Develop and scale generative models like diffusion and flow-matching for autonomous driving plan generation. Collaborate across teams to productize models for real-world deployment, requiring PhD/MSc + 3+ years in generative modeling and strong Python/C++ skills.

194k – 352kMountain View, CAAI ResearchOn-siteC++LLMs
Webflow

Senior Staff Machine Learning Scientist, Assets

Leads research in computer vision, multimodal understanding, and visual generation. Develops novel models and methodologies, translates research to production, and mentors teams. Requires PhD preferred, 8+ years experience, and expertise in PyTorch, TensorFlow, transformers.

194k – 285kUnited StatesAI ResearchRemote8+ YOEPythonPyTorch
Polymath

Member of Technical Staff - Research

Conducts applied research on long-horizon autonomous AI agents, focusing on evaluation, post-training, environment design, and benchmarks to improve frontier models. Builds simulations, runs experiments, ships production code, and publishes findings.

200k – 350kSan Francisco, CAAI ResearchOn-sitePythonAI Agents
Machinify

Staff NLP Scientist

Develops state-of-the-art VQA systems for medical records using advanced NLP and CV techniques to achieve expert-level accuracy in document analysis. Requires deep expertise in NLP/CV, strong software engineering, and ability to work with noisy data in healthcare domain.

180k – 280kUnited StatesAI ResearchRemoteNLPPython