Skip to content
DrataDrataSan Francisco, CA

Senior Applied Research Engineer 2

Drive AI system quality through experimentation and applied research on RAG, agents, and retrieval systems for Drata's compliance platform. Own evaluation frameworks, prototype GenAI workflows, and collaborate with engineers to productionize validated approaches.

192k – 260k/yr
Hybrid6+ YOEML Engineering

About the role

What you'll do

  • 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

What you'll bring

  • 6+ 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
  • Bonus: Experience with compliance, legal, or document-heavy domains
  • Bonus: Publications or contributions in IR, NLP, or RAG evaluation

Compensation & Benefits

  • Competitive base salary range: $192,000 - $259,800
  • 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, Kindbody fertility benefits
  • Generous annual professional and personal development stipends
  • Flexible vacation policy and paid holidays

Skills

PythonRAGInformation RetrievalNLPVector DatabasesEmbedding ModelsEvaluation FrameworksA/B TestingCross-EncodersLearning-To-Rank

Similar roles

ML Engineering jobs
Instacart

Senior Software Engineer II, ML/AI Platform

InstacartUnited States

As a Senior Software Engineer II on the ML/AI Platform team, you will build and define the internal platform for training and deploying AI models across the organization. This role involves developing SDKs and supporting infrastructure for AI model fine-tuning and batch inference.

192k – 242k/yr
Remote3+ YOEML Engineering
Instacart

Senior Software Engineer II, AI Labs & Foundations

InstacartUnited States

Instacart is seeking a Senior Software Engineer II for their AI Labs & Foundations team to design, build, and operate high-scale production AI systems. This role involves working on cutting-edge AI experiences like conversational shopping agents and voice AI, requiring expertise in robust software engineering and production AI/ML.

192k – 242k/yr
Remote5+ YOEML Engineering
Zoox

Senior AI Developer Productivity Engineer

ZooxFoster City, CA

Develops LLM-powered tools to boost developer productivity 10x, including code assistance, automated reviews, and task automation for autonomy software. Requires 7+ years experience as full-stack/backend developer fluent in Python or C++ with LLM optimization knowledge.

192k – 300k/yr
On-site7+ YOEML Engineering
Airbnb

Senior Machine Learning Engineer, Relevance and Personalization

AirbnbUnited States

Senior ML Engineer on Airbnb's Relevance and Personalization team building end-to-end ranking algorithms, models, and infrastructure for search, recommendations, and marketplace optimization using large-scale structured and unstructured data.

191k – 225k/yr
Remote5+ YOEML Engineering
Illumio

Sr. Machine Learning Engineer

IllumioSunnyvale, CA

Architects high-scale distributed systems and agentic AI for cybersecurity platform, processing massive data with Kafka, Spark/Flink, and Kubernetes. Requires 5-8 years backend experience in Java/Python/Go and expertise in agentic frameworks, RAG, and cloud infrastructure.

191k – 220k/yr
On-site5+ YOEML Engineering