GrepJob
Drata

Staff Applied AI Engineer

Drata
Apply
2 months ago
Remote, United States +3 moreStaff+
H1B Sponsor

Base Salary

$177k - $299k/yr

Responsibilities

  • Design and evaluate information access and reasoning strategies.
  • Prototype GenAI workflows for compliance objects.
  • Explore ML and probabilistic approaches for AI optimization.
  • Build and maintain evaluation frameworks and quality metrics.
  • Implement and tune ranking and reranking systems.
  • Run experiments to validate hypotheses and quantify improvements.
  • Debug failure modes and build error taxonomies.
  • Collaborate with AI and Software Engineers for production handoff.
  • Stay current on applied research in AI and compliance.

Requirements

  • 6+ years of experience in applied research, data science, or ML.
  • 2+ years of hands-on experience with production AI/ML systems.
  • Strong foundation in information retrieval and search relevance.
  • Experience with RAG systems and retrieval optimization.
  • Proficiency in evaluation methodology and statistical analysis.
  • Strong Python skills and experience with notebook-driven research.
  • Ability to communicate research findings to engineering teams.
  • Bonus: Experience in compliance or document-heavy domains.
  • Bonus: Publications in IR, NLP, or RAG evaluation.

Benefits

  • Stock equity to share in the company's success.
  • Up to 100% employer-paid premiums for health coverage.
  • Comprehensive financial benefits including a 401(k) plan.
  • Paid Parental Leave and family-building benefits.
  • Generous stipends for professional and personal development.
  • Flexible vacation policy and paid holidays.

Tech Stack

Categories

AI & MLData Science