Drata
Staff AI Engineer
Remote - US · Staff+
Sponsorship not specified$201k-$272kDetected 141 days ago
TypeScriptPythonSnowflakeVector DatabasesTemporalRESTMachine LearningLLMsRAGLLMOpsAI OrchestrationResearchCommunication
About the role
- 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.
- You'll help define how AI is architected across the company.
Responsibilities
- Design and own production AI systems end-to-end (LLM pipelines, RAG, reranking, vector stores, orchestration).
- Make thoughtful build/don't-build decisions based on real data.
- Design evaluation systems that measure retrieval quality, reasoning accuracy, end-to-end performance.
- Build tooling that helps the team iterate confidently and catch regressions early.
- Mentor senior engineers through design reviews and hands-on collaboration.
- Partner with product and compliance teams to translate domain complexity into clean technical solutions.
Requirements
- Deep experience with RAG, embeddings, reranking, vector databases (Pinecone, FAISS, Chroma, etc.), and agentic workflows.
Nice to have
- Experience in compliance, security, or other regulated domains.
- Familiarity with enterprise data platforms or Snowflake-based analytics.
- Experience with orchestration systems like Temporal or Airflow.
- Contributions to technical communities or published work.
- Why this role matters:
- You won't just ship features - 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.
Skills
- Experience designing evaluation frameworks and using quantitative analysis to improve system performance.
Compensation
- $200,700 - $271,500, subject to change.
- This role will receive a competitive base salary, benefits, and stock, typically in the form of Restricted Stock Units (RSUs).
Benefits
- We provide stock equity to ensure that as the company grows, you share directly in that success.
- Equity gives every employee a sense of ownership and the opportunity to celebrate our wins together-because your contributions don't just support our progress; they help drive our collective success.
- We want to support you in life's most important moments, so we offer a paid Parental Leave policy, after six months of employment.
- Employees also receive access to Kindbody fertility and family-building benefits and dedicated leave specialists who help guide you through the entire process.
- Drata offers a flexible vacation policy, paid holidays, and other perks to recharge.
Company info
- Hear the Voice of the Team https://drata.com/about/life-at-drata: Explore our "Life at Drata" page for employee testimonials on our collaborative and the growth opportunities available.
- Experience the Impact https://www.greatplacetowork.com/certified-company/7044563: See why we are consistently recognized on Fortune's Best Workplaces lists.
- A comprehensive suite of financial benefits, including a 401(k) plan, company-paid life and disability insurance, tax-advantaged spending accounts, and a range of discounted voluntary offerings to help you customize and strengthen your overall financial position.
This listing is sourced directly from Drata's careers page and normalized into a canonical job model.