Senior AI Product Engineer, Audit

Drata

Senior AI Product Engineer responsible for full-stack AI feature development within compliance software. Collaborating with product teams to enhance user experience using advanced AI technologies.

Posted 5/26/2026full-timeSan Francisco • California • 🇺🇸 United StatesSenior💰 $192,000 - $259,800 per yearWebsite

Tech Stack

Tools & technologies

JavaScriptNode.jsPythonReactTypeScript

About the role

Key responsibilities & impact

Build AI-powered product features end-to-end — React/TypeScript UI through Node.js/Python backend — with real-time streaming, graceful degradation, and human-in-the-loop interaction patterns
Translate AI capabilities — RAG pipelines, agentic workflows, structured reasoning — into interactions that feel natural to compliance practitioners
Partner with AI Engineers to define API contracts and output schemas
Advocate for the user's perspective in technical decisions; surface where model outputs break down in practice and iterate the product layer accordingly
Build user-facing feedback loops that capture signal on AI output quality and make it actionable for the broader AI team
Instrument AI features with product-level observability — latency, engagement, task completion, drop-off — alongside integration and end-to-end tests that validate behavior across the full stack
Establish reusable patterns (streaming hooks, feedback components, AI state management) that accelerate future AI feature development
Mentor engineers newer to AI product development; participate in design reviews with both engineering depth and product instinct

Requirements

What you’ll need

7+ years of software engineering experience with a proven track record of shipping full-stack features in production
2+ years working on AI-powered product features
Strong proficiency in React and TypeScript; experience building responsive, interactive UIs that handle async, streaming, and real-time data gracefully
Solid Python and Node.js and Typescript skills; experience designing and consuming APIs that integrate with LLM providers, orchestration frameworks, or AI services
Hands-on experience building with LLM APIs (OpenAI, Anthropic, etc.); practical understanding of prompt engineering, context management, RAG, Quality/Evals(Braintrust or others) and structured outputs
Familiarity with agent frameworks (LangGraph, LangChain, custom runtimes) and how to build product surfaces on top of multi-step agentic workflows
Experience implementing streaming response patterns (SSE, WebSockets, chunked HTTP) and building UIs that handle progressive, partial, and uncertain outputs
Strong intuition for user experience; you ask 'does this feel right to the user' as often as 'does this work technically' — and you're right about both
Track record of working closely with AI/ML engineers, product managers, and designers; comfortable operating in a cross-functional product team where the requirements are often ambiguous and the iteration is fast

Benefits

Comp & perks

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, and a range of discounted voluntary offerings to help you customize and strengthen your overall financial position
Paid Parental Leave policy, after six months of employment
Access to Kindbody fertility and family-building benefits and dedicated leave specialists who help guide you through the entire process
Generous annual stipends for both professional and personal development
Flexible vacation policy, paid holidays, and other perks to recharge

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

ReactTypeScriptPythonNode.jsAPI designLLM APIsprompt engineeringstreaming response patternsuser experiencefull-stack development

Soft Skills

mentoringcross-functional collaborationuser advocacyiterative product developmentdesign review participationproblem-solvingcommunicationadaptabilityintuitive thinkinguser-centered design