Senior AI Engineer, Agent Harness

Drata, Inc.

Job Overview

Location

Hybrid - San Francisco

Job Type

Full-time

Full Job Description

📋 Description

• Design and implement LLM-powered agentic systems capable of multi-step reasoning, evidence grounding, and decision support in high-stakes compliance environments.
• Develop agentic workflows that integrate retrieval, tool use, structured reasoning, and human oversight to automate compliance and risk assessment tasks.
• Build interactive AI experiences that enable users to naturally engage with complex compliance, security, and regulatory data through intuitive interfaces.
• Construct AI systems that reason over structured and unstructured data to interpret regulations, validate controls, and assess risk while ensuring outputs are traceable, explainable, and auditable.
• Architect, deploy, and optimize scalable production-grade LLM systems with retrieval-augmented generation (RAG), vector databases, and agent frameworks, prioritizing latency, cost, reliability, and evaluation metrics.
• Partner with platform, security, product, and engineering teams to operationalize AI features safely and effectively within Drata’s trust platform.
• Embed human-in-the-loop workflows, confidence thresholds, and safety guardrails into AI systems to ensure responsible deployment in regulated environments.
• Implement privacy-preserving data handling practices and robust failure modes that align with Drata’s values of integrity, trust, and customer obsession.
• Own the end-to-end lifecycle of AI-driven features—from concept and prototyping to production deployment and iterative improvement—across the Drata platform.
• Ensure all AI systems meet enterprise compliance standards by maintaining transparency, auditability, and alignment with regulatory frameworks such as SOC 2, ISO 27001, and GDPR.
• Collaborate with non-technical stakeholders including compliance, GRC, and audit teams to translate domain-specific requirements into technical AI solutions.
• Contribute to setting the technical and ethical standard for applying modern AI in trust-critical, regulated enterprise environments.
• Drive innovation in automated reasoning over regulations and evidence, enabling customers to derive accurate, actionable insights from complex compliance data.
• Support the evolution of foundational agentic AI capabilities that power multiple current and future product experiences at Drata.

🎯 Requirements

• 5+ years of hands-on software engineering experience
• 2+ years specifically in ML/AI engineering
• Proficiency in Python
• Proven experience building and shipping LLM-based applications in production, including embeddings, RAG, agent frameworks, and prompt engineering
• Familiarity with vector databases (Pinecone, Chroma, FAISS, etc.) and RAG system design
• Track record of taking AI systems from concept to production with scalable, maintainable architecture

🏖️ Benefits

• Stock equity in the form of Restricted Stock Units (RSUs)
• Up to 100% employer-paid premiums for medical, dental, and vision coverage for employees and dependents
• Comprehensive wellness benefits and healthcare concierge services
• 401(k) plan, company-paid life and disability insurance, and tax-advantaged spending accounts
• Paid Parental Leave after six months of employment and access to Kindbody fertility and family-building benefits
• Generous annual stipends for professional and personal development
• Flexible vacation policy and paid holidays

Skills & Technologies

Python

TypeScript

REST

Data Science

Senior

Hybrid

$166k-225k

Ready to Apply?

Apply Externally

You will be redirected to an external site to apply.

AI Job Fit Analysis

Pro

See exactly how your profile matches this role — strengths, skill gaps, and what to do about them.

Drata, Inc.

Visit Website

About Drata, Inc.

Drata is a compliance automation platform that continuously monitors and collects evidence for security frameworks such as SOC 2, ISO 27001, HIPAA, and PCI DSS. It integrates with cloud services, identity providers, and SaaS tools to track controls, manage risks, and generate audit-ready documentation. The company serves technology startups and enterprises seeking to streamline security compliance workflows and maintain continuous monitoring without manual spreadsheets.

View Company Profile

Get more remote jobs like this

Subscribe to the weekly newsletter for similar remote roles and curated hiring updates.

Weekly remote jobs and featured talent.

No spam. Only curated remote roles and product updates. You can unsubscribe anytime.