Senior AI Product Engineer 2, Control Remediation

Drata is building the trust layer between great companies - automating compliance, managing risk, and helping organizations prove trust continuously as they scale. We're Dratanauts: a global crew of 600+ professionals united by a culture that rewards integrity, ownership, and raising the bar, no matter where in the world we're working from.

Why Join the Drata Team? At Drata, you're not maintaining legacy compliance software - you're building the agentic AI platform defining what trust looks like for the next generation of companies. Here's what makes the work itself worth showing up for:

• Problems without a playbook: You'll work at the edge of AI and security, building agentic governance, continuous compliance, and real-time trust verification to solve problems that don't have an established answer yet. You're writing it as you go.
• Real ownership, not just process: Our values center on owning outcomes and raising the bar, not checking boxes. You're expected to have opinions and back them.
• A seat at the table: Your perspective is unique and valued. Open debate and diverse viewpoints are built into how decisions actually get made here, at every level.
• Growth at rocketship speed: Drata is scaling fast, which means scope grows fast too. High performers get more ownership, visibility, and experience.
• A crew, not just coworkers: Dratanauts consistently describe a "come as you are" culture with sharp, curious people—the kind of team that makes hard problems genuinely fun to solve.linkedin.com/company/drata/posts/?feedView=all for company news, employee stories, and career updates.

Job Summary:

Drata is reimagining compliance as an intelligent, always-on experience — and AI is at the center of that vision. We are seeking a Senior AI Product Engineer to own the full-stack development of customer-facing AI features, embedded directly within our product teams. This is not a platform or infrastructure role. You'll translate the capabilities of LLMs, agents, and RAG pipelines into intuitive, polished product experiences — streaming chat interfaces, agentic workflows, intelligent summaries, and guided automation that makes compliance feel effortless. You'll partner closely with AI Engineers who own the backend intelligence and Product and Design who shape the vision, but you are the engineer who closes the loop between AI capability and the customer experience.

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 p