# Senior AI Product Engineer 2, Evidence

**Company:** [Drata](https://hotfix.jobs/companies/drata)
**Location:** San Francisco, CA
**Role:** ML Engineering
**Experience:** 7+ years
**Skills:** React, TypeScript, Node.js, Python, LLM APIs, Prompt Engineering, RAG, LangGraph, LangChain, Sse, WebSockets
**Posted:** 2026-05-26

> Drata is seeking a Senior AI Product Engineer to lead full-stack development of customer-facing AI features. This role involves translating LLM capabilities into intuitive product experiences, partnering with AI Engineers, and advocating for user experience in technical decisions.

## Job Description

## What you'll do:
* 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; translate RAG pipelines, agentic workflows, and structured reasoning into interactions that feel natural to compliance practitioners
* 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

## What you'll bring:
* **Experience:** 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
* **Frontend:** Strong proficiency in React and TypeScript; experience building responsive, interactive UIs that handle async, streaming, and real-time data gracefully
* **Backend:** Solid Python and Node.js and Typescript skills; experience designing and consuming APIs that integrate with LLM providers, orchestration frameworks, or AI services
* **AI Fluency:** 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 — not just calling an endpoint, but knowing what makes a prompt reliable in production
* Familiarity with agent frameworks (LangGraph, LangChain, custom runtimes) and how to build product surfaces on top of multi-step agentic workflows
* **Streaming & Real-Time UI:** Experience implementing streaming response patterns (SSE, WebSockets, chunked HTTP) and building UIs that handle progressive, partial, and uncertain outputs
* **Product Sense:** 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
* **Collaboration:** 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

## Bonus Qualifications
* Experience in compliance, security, GRC, or other document-heavy enterprise domains
* Experience with MCP (Model Context Protocol) or similar tool-use interfaces for AI agents
* Exposure to RAG system design — not as an owner, but as a consumer who has debugged retrieval quality issues from the product layer
* Experience building feedback and annotation tooling that captures user signal on AI output quality

## Impact
This role is the bridge between Drata's AI ambitions and the customers who will benefit from them. Your work will:
* Define the standard for how AI features look, feel, and behave across Drata's product — setting a bar that the entire product engineering organization will build to
* Make compliance practitioners more capable and confident by turning AI intelligence into product interactions they actually trust and use
* Accelerate the velocity of AI product development by establishing reusable patterns, components, and interfaces that make the next AI feature faster to ship than the last
* Influence the roadmap through your proximity to both the customer experience and the AI capability layer — you'll see the gaps and the opportunities before almost anyone else

## How we support you:
At Drata, our people are our strongest advantage—and we prove it with support that exceeds industry standards. Our total rewards package is designed to power your well-being, accelerate your growth, and keep your work-life balance thriving.

Explore how we invest in your Life at Drata.

**Shared Success:** We provide stock equity to ensure that as the company grows, you share directly in that success. Equity gives every employee a sense of ownership and the opportunity to celebrate our wins together—because your contributions don’t just support our progress; they help drive our collective success.

**Health & Wellness:** Up to 100% employer-paid premiums for medical, dental, and vision coverage for employees and their dependents, along with comprehensive wellness benefits and healthcare concierge services designed to support your needs beyond traditional insurance.

**Financial Well-being:** A comprehensive suite of financial benefits, including a...

## Similar roles

- [Senior Machine Learning Engineer, Agent Oversight](https://hotfix.jobs/jobs/1f504a69-044f-47ab-872b-120a1706cd88) - Scale AI - San Francisco, CA - $216k – $270k/yr
- [Senior Software Engineer, Agent Oversight](https://hotfix.jobs/jobs/808ce719-6b16-4b6e-8887-8c8f8a780b75) - Scale AI - San Francisco, CA - $216k – $270k/yr
- [Senior Software Engineer, AI Infrastructure](https://hotfix.jobs/jobs/dd1c3420-6c0a-47ed-a1a7-de401bdf506c) - Ambient.ai - Redwood City, CA - $168k – $205k/yr
- [Senior Software Engineer](https://hotfix.jobs/jobs/423c60aa-ca4c-400d-babe-a50099433cbb) - SpotOn - Chicago, IL - $160k – $190k/yr
- [Senior AI Engineer, Enterprise](https://hotfix.jobs/jobs/87a30fd1-3fe6-47f3-a0a1-da444f3c00f4) - Snowflake - Menlo Park, CA - $156k – $224k/yr

**Apply:** https://hotfix.jobs/jobs/88aaef30-6312-41d2-89b7-fadcb5146212
**Canonical:** https://hotfix.jobs/jobs/88aaef30-6312-41d2-89b7-fadcb5146212