Senior AI Product Engineer, Frameworks

Drata

. 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 .

Posted 5/26/2026full-timeSan Francisco • 🇺🇸 United StatesSenior💰 $166,700 - $225,900 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
Establish reusable patterns (streaming hooks, feedback components, AI state management) that accelerate future AI feature development
Mentor engineers newer to AI product development

Requirements

What you’ll need

5+ 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)
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

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
A comprehensive suite of financial benefits, including a 401(k) plan, company-paid life and disability insurance, tax-advantaged spending accounts, and discounted voluntary offerings
Paid Parental Leave policy after six months of employment
Access to Kindbody fertility and family-building benefits
Generous annual stipends for both professional and personal development
Access to a wide range of internal learning opportunities
Flexible vacation policy, paid holidays, and other perks

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

mentoringuser advocacycross-functional collaborationproblem-solvingiterative developmentcommunicationintuitive designfeedback incorporationtechnical decision-makinguser perspective