Skip to content
DrataDrataSan Francisco, CA

Senior Applied Research Engineer

Senior Applied Research Engineer driving AI system quality through experimentation and evaluation of RAG, retrieval, and reasoning systems. Requires 5+ years applied ML/NLP experience with strong Python and evaluation methodology skills.

167k – 226k
Hybrid5+ YOEML Engineering

About the role

Responsibilities

  • Design and evaluate information access + reasoning strategies across RAG, agents, and classic ML: chunking, embedding models, hybrid search, metadata filtering, semantic routing
  • Prototype GenAI workflows (including agentic systems) that map and reason over compliance objects (controls ↔ risks ↔ requirements ↔ evidence)
  • Explore ML + probabilistic approaches where GenAI is not the best fit: classifiers, ranking models, graph/link prediction, calibration, and structured prediction
  • Build and maintain evaluation frameworks: golden datasets, automated quality metrics, regression detection
  • Implement and tune ranking/reranking systems: cross-encoders, LLM-based rerankers, learning-to-rank, custom scoring functions
  • Run experiments to validate hypotheses and quantify improvements before production rollout
  • Debug failure modes and build error taxonomies across retrieval, reasoning, and generation
  • Collaborate with AI and Software Engineers to hand off validated approaches for productionization
  • Stay current on applied research in RAG, agents, LLM evaluation, and relevance modeling; bring innovations into the product

Requirements

  • 5+ years of experience in applied research, data science, or ML with a focus on NLP, information retrieval, or knowledge systems
  • 2+ years of hands-on experience building or contributing to production AI/ML systems
  • Strong foundation in information retrieval: dense and sparse retrieval, embedding models, search relevance
  • Experience with RAG systems: chunking strategies, vector databases, retrieval optimization
  • Proficiency in evaluation methodology: metrics design, golden dataset creation, A/B testing, statistical significance
  • Strong Python skills and comfort with notebook-driven research workflows
  • Experience communicating research findings to engineering teams and translating insights into actionable improvements

Nice-to-Haves

  • Experience with compliance, legal, or document-heavy domains
  • Publications or contributions in IR, NLP, or RAG evaluation

Compensation & Benefits

  • Competitive base salary: $166,900 - $225,900
  • Stock equity (RSUs)
  • Up to 100% employer-paid medical, dental, and vision premiums
  • 401(k) plan, company-paid life and disability insurance
  • Paid Parental Leave (after 6 months)
  • Kindbody fertility and family-building benefits
  • Generous annual professional and personal development stipends
  • Flexible vacation policy and paid holidays

Skills

PythonRAGInformation RetrievalNLPMachine LearningVector DatabasesEvaluation MetricsA/B TestingStatistical AnalysisEmbedding Models

Similar roles

ML Engineering jobs
Drata

Senior AI Engineer, Agent Harness

DrataSan Francisco, CA

Senior AI Engineer to design, build, and scale agentic AI systems using LLMs for compliance automation. Own end-to-end development of production LLM + retrieval + agent workflows with focus on responsible AI.

167k – 226k
Hybrid5+ YOEML Engineering
Mercury

Senior Machine Learning Operations Engineer

MercurySan Francisco, CA +2

Build and operate Mercury's real-time ML inference platform for fraud risk decisioning. Own model deployment, observability, and lifecycle tooling with strong backend Python fundamentals.

167k – 208k
Hybrid5+ YOEML Engineering
Databricks

Senior Software Engineer, Model Serving

DatabricksSan Francisco, CA

Designs and builds scalable infrastructure for high-throughput, low-latency AI/ML model serving on CPU/GPU. Requires 5+ years in distributed systems, inference expertise, and strong system design skills.

166k – 225k
On-site5+ YOEML Engineering
Databricks

Senior Machine Learning Engineer - GenAI Platform

DatabricksSan Francisco, CA

Build customer-facing generative AI platform covering ML lifecycle from data generation to agent-building. Requires 4+ years experience in distributed systems and ML platforms, with strong product ownership.

166k – 225k
On-site4+ YOEML Engineering
Ambient.ai

Senior Software Engineer, AI Infrastructure

Ambient.aiRedwood City, CA

Build and optimize scalable AI infrastructure for real-time inference, evaluation, and continuous improvement of LLMs, LVMs, computer vision, and multimodal models on large-scale video data. Requires 4+ years production ML systems experience, strong Python skills, and expertise in inference optimization and model serving.

168k – 205k
Hybrid4+ YOEML Engineering