Skip to content
DrataDrataSan Francisco, CA

Applied Research Engineer

Drive the quality of Drata's AI systems through experimentation and applied research on RAG, agents, and information retrieval. Own evaluation frameworks and translate validated approaches into production with AI and Software Engineers.

145k – 196k/yr
Hybrid3+ 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

  • 3+ years of experience in applied research, data science, or ML with a focus on NLP, information retrieval, or knowledge systems
  • 1+ 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
  • Publications or contributions in IR, NLP, or RAG evaluation

Compensation & Benefits

  • Competitive base salary: $145,200 - $196,400
  • 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 RetrievalNLPEmbedding ModelsVector DatabasesEvaluation MetricsA/B TestingCross-EncodersLearning-To-Rank

Similar roles

ML Engineering jobs
Databricks

Software Engineer - GenAI inference

DatabricksSan Francisco, CA

Designs, develops, and optimizes GenAI inference engine for LLMs, focusing on performance, scalability, and efficiency across the full stack. Requires 3+ years in performance-critical systems, CUDA/GPU expertise, and ML inference knowledge.

142k – 205k/yr
On-site3+ YOEML Engineering
Figma

Software Engineer, AI Platforms

FigmaSan Francisco, CA +1

Build scalable AI platforms and infrastructure for Figma's design tools, including model training, agentic features, and APIs. Requires 5+ years software engineering experience with backend/infrastructure and 3+ years in AI or developer platforms.

149k – 350k/yr
Remote5+ YOEML Engineering
Figma

Software Engineer, Machine Learning

FigmaSan Francisco, CA +1

Build and productionize ML models for search, RAG, and generative AI features at Figma. Requires 5+ years software engineering with 3+ years in applied ML, Python proficiency, and experience with scalable data pipelines.

149k – 350k/yr
Remote5+ YOEML Engineering
Applied Intuition

Perception & Fusion Engineer

Applied IntuitionArlington, VA

Develop and integrate real-time sensor perception and fusion algorithms for autonomous vehicles across land, air, sea, and space domains. Requires MS/PhD or 5+ years experience with multi-modal sensors (EO/IR/radar), ML deployment, and DoD clearance eligibility.

150k – 220k/yr
On-site5+ YOEML Engineering
Deepgram

Applied ML Engineer

DeepgramCalifornia

Own the research-to-production pipeline at Deepgram, turning experimental speech ML models into reliable, scalable production services. Partner with researchers on robust workflows, automated release gates, inference optimization, and feedback loops across hybrid GPU infrastructure.

150k – 220k/yr
Remote5+ YOEML Engineering