[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.ResponsibilitiesDesign and implement LLM-powered systems capable of multi-step reasoning, evidence grounding, and decision support in high-trust domainsDevelop agentic workflows that combine retrieval, tool use, structured reasoning, and human oversightCreate interactive AI experiences that allow users to engage naturally with complex compliance and risk dataBuild AI systems that reason over structured and unstructured data to support regulatory interpretation, control alignment, and evidence validationEnsure AI outputs are traceable, explainable, and auditable, meeting the expectations of enterprise customers and auditorsArchitect and deploy scalable LLM + retrieval + agent systems in production environmentsOptimize for latency, cost, reliability, and evaluation in real-world enterprise workloadsPartner with platform, security, product teams, and other application development teams with diverse skill sets to operationalize AI safely and effectivelyEmbed human-in-the-loop workflows, confidence thresholds, and safety guardrails into AI systemsEnsure privacy-preserving data handling, robust failure modes, and transparent behavior aligned with Drata’s trust principlesSkills7+ years of hands-on software engineering experience2+ years specifically in ML/AI engineeringProficiency in PythonFamiliarity with vector databases (Pinecone, Chroma, FAISS, etc.) and RAG system designProven experience building and shipping LLM-based applications in production, including embeddings, RAG, agent frameworks, prompt engineering, and evaluationTrack record of taking AI systems from concept to production, designing scalable, maintainable solutionsAbility to decompose complex tasks into agentic workflows and reason over high-stakes, structured, and unstructured dataExperience working cross-functionally with product, compliance, security, engineering teams, and partnering with other application development teams with diverse skill setsExperience in compliance, security, risk, or audit domainsFamiliarity with Snowflake-based analytics, knowledge graphs, or enterprise data platformsExperience partnering with non-technical stakeholders such as compliance or GRC teamsBenefitsStock equity to ensure that as the company grows, you share directly in that successUp to 100% employer-paid premiums for medical, dental, and vision coverage for employees and their dependentsComprehensive wellness benefits and healthcare concierge services designed to support your needs beyond traditional insuranceA comprehensive suite of financial benefits, including a 401(k) planCompany-paid life and disability insuranceTax-advantaged spending accountsA range of discounted voluntary offerings to help you customize and strengthen your overall financial positionA paid Parental Leave policy, after six months of employmentEmployees also receive access to Kindbody fertility and family-building benefitsDedicated leave specialists who help guide you through the entire processGenerous annual stipends for both professional and personal developmentAccess to a wide range of internal learning opportunitiesA flexible vacation policyPaid holidaysOther perks to rechargeCompany OverviewDrata is a Trust Management Platform that automates compliance workflows, streamlines risk management, and continuous security assurance It was founded in 2020, and is headquartered in San Diego, California, USA, with a workforce of 501-1000 employees. Its website is https://drata.com.Company H1B SponsorshipDrata has a track record of offering H1B sponsorships, with 1 in 2026, 1 in 2025, 2 in 2024, 1 in 2023. Please note that this does not guarantee sponsorship for this specific role.