[Remote] Senior AI Engineer

Note: The job is a remote job and is open to candidates in USA. 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

• Design and implement LLM-powered systems capable of multi-step reasoning, evidence grounding, and decision support in high-trust domains
• 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
• 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
• 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
• 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

Skills

• 7+ years of hands-on software engineering experience
• 2+ years specifically in ML/AI engineering
• Proficiency in Python
• 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, and evaluation
• 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, engineering teams, and partnering with other application development teams with diverse skill sets
• 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

Benefits

• Stock equity to ensure that as the company grows, you share directly in that success
• Up to 100% employer-paid premiums for medical, dental, and vision coverage for employees and their dependents
• Comprehensive wellness benefits and healthcare concierge services designed to support your needs beyond traditional insurance
• A comprehensive suite of financial benefits, including a 401(k) plan
• Company-paid life and disability insurance
• Tax-advantaged spending accounts
• A range of discounted voluntary offerings to help you customize and strengthen your overall financial position
• A paid Parental Leave policy, after six months of employment
• Employees also receive access to Kindbody fertility and family-building benefits
• Dedicated leave specialists who help guide you through the entire process
• Generous annual stipends for both professional and personal development
• Access to a wide range of internal learning opportunities
• A flexible vacation policy
• Paid holidays
• Other perks to recharge

Company Overview

Company H1B Sponsorship