Senior AI Engineer

About the role

Drata is advancing the frontier of compliance automation by integrating intelligent AI capabilities into its platform. We are seeking a Senior AI Engineer to help design, build, and scale robust, high-impact AI systems that improve operational efficiency, enhance decision-making, and support trust-critical enterprise workflows.

Responsibilities

• Build Agentic & Intelligent AI Systems
• Design and implement LLM-powered systems capable of multi-step reasoning, evidence grounding, and decision support in high-trust domains.
• Create agentic workflows that combine retrieval, tool use, structured reasoning, and human oversight.
• Create interactive AI experiences that allow users to engage naturally with complex compliance and risk data.
• Automated Reasoning Over Regulations & Evidence
• Build AI systems that reason over structured and unstructured data to support regulatory interpretation, control alignment, and evidence validation.
• Ensure AI outputs are traceable, explainable, and auditable, meeting the expectations of enterprise customers and auditors.
• Production-Grade AI Architecture
• Architect and deploy scalable LLM + retrieval + agent systems in production environments.
• Optimize for latency, cost, reliability, and evaluation in real-world enterprise workloads.
• Partner with platform, security, product teams, and other application development teams with diverse skill sets to operationalize AI safely and effectively.
• Responsible & Trustworthy AI
• Embed human-in-the-loop workflows, confidence thresholds, and safety guardrails into AI systems.
• Ensure privacy-preserving data handling, robust failure modes, and transparent behavior aligned with Drata’s trust principles.

Requirements

Experience: 7+ years of hands-on software engineering experience; 2+ years specifically in ML/AI engineering.

• Programming Languages: Proficiency in Python; TypeScript experience is a plus, especially for production AI system integrations.
• Retrieval & Vector Stores: Familiarity with vector databases (Pinecone, Chroma, FAISS, etc.) and RAG system design.
• LLM & AI Systems: Proven experience building and shipping LLM-based applications in production, including embeddings, RAG, agent frameworks, prompt engineering, and evaluation.
• Production Expertise: Track record of taking AI systems from concept to production, designing scalable, maintainable solutions.
• Problem Solving: Ability to decompose complex tasks into agentic workflows and reason over high-stakes, structured, and unstructured data.
• Collaboration: Experience working cross-functionally with product, compliance, security, engineering teams, and partnering with other application development teams with diverse skill sets.

Qualifications

• Experience in compliance, security, risk, or audit domains.
• Familiarity with Snowflake-based analytics, knowledge graphs, or enterprise data platforms.
• Experience partnering with non-technical stakeholders such as compliance or GRC teams.

Skills

• Strong programming skills in Python and TypeScript.
• Experience with vector databases and RAG systems.
• Proven track record in building and deploying LLM-based applications.
• Ability to design and implement scalable, maintainable AI solutions.
• Experience with automated reasoning and evidence validation.
• Knowledge of compliance, security, and risk domains.

Benefits

At Drata, our people are our strongest advantage—and we prove it with support that exceeds industry standards. Our total rewards package is designed to power your well-being, accelerate your growth, and keep your work-life balance thriving.

• Stock equity
• Up to 100% employer-paid premiums for medical, dental, and vision coverage for employees and their dependents
• Comprehensive wellness benefits and healthcare concierge services
• 401(k) plan and company-paid life and disability insurance
• Tax-advantaged spending accounts
• Discounted voluntary offerings
• Paid Parental Leave policy
• Kindbody fertility and family-building benefits
• Dedicated leave specialists
• Generous annual stipends for both professional and personal development
• Wide range of internal learning opportunities
• Flexible vacation policy, paid holidays, and other perks

Pay

A competitive base salary, benefits, and/or stock, typically in the form of Restricted Stock Units (RSUs).

Schedule

We believe that to do your best work, you should get the time you need for rest, rejuvenation and recovery. Drata offers a flexible vacation policy, paid holidays, and other perks to recharge.