Applied Research Engineer
Drata · San Francisco, CA · 3 wk ago
HybridEngineering$145k–$196k/yrFull-time
About the role
The Applied Research Engineer will drive the quality and effectiveness of Drata's AI systems through rigorous experimentation, evaluation, and applied research. This role emphasizes experimentation and rigor over production engineering.
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
- 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
Qualifications
- Experience with compliance, legal, or document-heavy domains
- Publishations or contributions in IR, NLP, or RAG evaluation
Skills
- Applied Research
- Data Science
- Machine Learning
- Natural Language Processing
- Information Retrieval
- Knowledge Systems
Benefits
- Stock Equity
- Health & Wellness
- Financial Well-being
- Family Support
- Growth & Development
- Time Off & Flexibility
Pay
$145,200 - $196,400
Schedule
Full-time