GrepJob
Drata

Staff AI Engineer

Drata
Apply
4 months ago
Remote, United StatesStaff+
H1B Sponsor

Base Salary

$201k - $272k/yr

Responsibilities

  • Design and own production AI systems end-to-end.
  • Make informed build/don't-build decisions based on data.
  • Evolve the AI stack from model infrastructure to evaluation tooling.
  • Design evaluation systems to measure performance and quality.
  • Build tooling for early regression detection.
  • Analyze production outputs to identify failure patterns.
  • Lead AI architecture decisions and mentor senior engineers.
  • Partner with product and compliance teams for technical solutions.
  • Ship systems optimized for latency, cost, and reliability.

Requirements

  • 10+ years of software engineering experience.
  • 3+ years working directly on ML/AI systems.
  • Experience with production LLM systems.
  • Deep knowledge of RAG, embeddings, and vector databases.
  • Strong Python skills; TypeScript is a plus.
  • Experience in designing evaluation frameworks.
  • Ability to analyze complex problems and provide clear recommendations.
  • Strong communication skills for cross-functional collaboration.

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

Apache AirflowPythonTypeScript

Categories

AI & MLData Science