Staff AI Engineer

Role Overview

At Drata, we're rethinking how compliance gets done, and AI is a core part of that. We're looking for a Staff AI Engineer to help shape how intelligent systems power trust-critical enterprise workflows.

What You Will Do

Design and own production AI systems end-to-end, make thoughtful build/don't-build decisions based on real data, evolve our AI stack over time, and raise the quality bar.

Why It Might Be a Fit

You'll help define how AI is built at Drata, set the architectural foundation for our AI systems, establish the quality bar that lets us move fast with confidence, and ensure our technical investments are grounded in evidence.

Requirements

• 10+ years of software engineering experience, including 3+ years working directly on ML/AI systems
• Real ownership of production LLM systems
• Deep experience with RAG, embeddings, reranking, vector databases, and agentic workflows
• Experience designing evaluation frameworks and using quantitative analysis to improve system performance
• Strong Python skills (TypeScript is a plus)
• A track record of making architectural decisions that shape team direction
• Experience operating AI systems in production - observability, reliability, cost tradeoffs
• The ability to break down ambiguous, high-stakes problems into structured investigations
• Clear communication skills and comfort working cross-functionally

Benefits

• Stock equity
• Up to 100% employer-paid premiums for medical, dental, and vision coverage
• Comprehensive wellness benefits and healthcare concierge services
• 401(k) plan
• Company-paid life and disability insurance
• Tax-advantaged spending accounts
• Paid Parental Leave policy
• Generous annual stipends for professional and personal development
• Flexible vacation policy
• Paid holidays