Senior AI Engineer, Agent Harness

Drata•San Francisco, CA

7d•$166,900 - $225,900•Hybrid

About The Position

Drata is advancing the frontier of compliance automation by integrating intelligent AI capabilities into its trust 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. This role focuses on solving complex, real-world problems with agentic AI, LLMs, and intelligent reasoning systems in a compliance and security context. As a Senior AI Engineer, you will own the end-to-end design, implementation, and evolution of AI-driven features across the Drata platform.

Requirements

5+ years of hands-on software engineering experience; 2+ years specifically in ML/AI engineering
Proficiency in Python; TypeScript experience is a plus, especially for production AI system integration
Familiarity with vector databases (Pinecone, Chroma, FAISS, etc.) and RAG system design
Proven experience building and shipping LLM-based applications in production, including embeddings, RAG, agent frameworks, prompt engineering
Track record of taking AI systems from concept to production, designing scalable, maintainable solutions
Ability to decompose complex tasks into agentic workflows and reason over high-stakes, structured, and unstructured data
Experience working cross-functionally with product, compliance, security, and engineering teams

Nice To Haves

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

Responsibilities

Build Agentic & Intelligent AI Systems: Design and implement LLM-powered systems capable of multi-step reasoning, evidence grounding, and decision support in high-stakes compliance environments.
Develop 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 validation, and risk assessment.
Ensure AI outputs are traceable, explainable, and auditable, meeting the expectations of enterprise compliance teams.
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 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 values.

Benefits

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
Company-paid life and disability insurance
Tax-advantaged spending accounts
Range of discounted voluntary offerings
Paid Parental Leave policy, after six months of employment
Kindbody fertility and family-building benefits
Dedicated leave specialists
Generous annual stipends for both professional and personal development
Access to a wide range of internal learning opportunities
Flexible vacation policy
Paid holidays