Skip to content

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 – 260kSan Francisco, CAML EngineeringHybrid6+ YOE

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

Senior Software Engineer II, ML/AI Platform

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 – 242kUnited StatesML EngineeringRemote3+ YOEGoC++

Senior Software Engineer II, AI Labs & Foundations

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 – 242kUnited StatesML EngineeringRemote5+ YOEAIRAG

Senior AI Developer Productivity Engineer

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 – 300kFoster City, CAML EngineeringOn-site7+ YOEC++LLMs

Sr. Machine Learning Engineer

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 – 220kSunnyvale, CAML EngineeringOn-site5+ YOEGoRAG

Senior Software Engineer, GenAI Platform

Leads development of Reddit's large-scale GenAI Platform, including LLM Gateway, RAG applications, and agentic AI workflows. Requires 5+ years in ML/AI platform engineering with expertise in cloud, Kubernetes, and MLOps practices.

191k – 267kUnited StatesML EngineeringRemote5+ YOEGoAWS