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 – 270k
On-site2+ YOEAI Research
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
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 – 260k
On-site7+ YOEAI Research
Senior/Staff Machine Learning Research Scientist: Generative Modeling for Planning
NuroMountain View, CA
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 – 352k
On-siteAI Research
Senior Staff Machine Learning Scientist, Assets
WebflowUnited States
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 – 285k
Remote8+ YOEAI Research
Member of Technical Staff - Research
PolymathSan Francisco, CA
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 – 350k
On-siteAI Research
Staff NLP Scientist
MachinifyUnited States
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.