Senior AI Engineer
About The Position
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. This role focuses on solving complex, real-world problems with agentic AI, LLMs, and intelligent reasoning systems, including automating questionnaire answering and other compliance-related workflows. You will work on creating generalizable AI solutions that can be applied across multiple products and domains, collaborating with cross-functional teams to bring cutting-edge research into production-ready systems. As a Senior AI Engineer, you will own the end-to-end design, implementation, and evolution of AI-driven systems embedded directly into Drata’s core platform. Your work will focus on building generalizable, reusable AI capabilities rather than one-off features, enabling multiple product surfaces and workflows across compliance and security.
Requirements
- 7+ 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 integrations.
- 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.
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
- 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.
Benefits
- Stock equity (RSUs)
- 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
- 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 professional and personal development
- Access to a wide range of internal learning opportunities
- Flexible vacation policy
- Paid holidays
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
Education Level
No Education Listed
