Senior AI Product Engineer - Real-Time UI & RAG

Cacheflow; San Francisco (CA); On-site; USD 192,000 - 260,000; Full time

Benefits offered by this job

Stock equity

Health & Wellness

401(k) plan

Parental Leave

Growth stipend

Flexible vacation

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 Product Engineer to own the full-stack development of customer-facing AI features, embedded within our product teams.

You'll translate LLMs, agents, and RAG pipelines into polished product experiences—streaming chat, agentic workflows, intelligent summaries, and guided automation—while partnering with Platform, Product, and Design to connect AI capability to

Qualifications

• 7+ years of software engineering experience
• 2+ years shipping AI-powered product features
• Proficiency in React and TypeScript

Responsibilities

• Build AI-powered product features end-to-end — React/TypeScript UI through Node.js/Python backend
• Translate AI capabilities — RAG pipelines, agentic workflows, structured reasoning — into natural interactions for compliance practitioners
• Define API contracts and output schemas; translate AI pipelines into product features
• Advocate for the user’s perspective in technical decisions and iterate the product layer accordingly
• Build user-facing feedback loops to capture AI output quality signals
• Instrument AI features with observability and tests across the full stack
• Establish reusable patterns to accelerate future AI development
• Mentor engineers newer to AI product development and participate in design reviews

Skills

React

TypeScript

Node.js

Python

API design

AI/ML

Streaming UI

Team collaboration

Tools

LangGraph

LangChain

LLM APIs

WebSockets

SSE

Job description

Drata is reimagining compliance as an intelligent, always-on experience—and AI is at the center of that vision. We are seeking a Senior Product Engineer to own the full-stack development of customer-facing AI features, embedded within our product teams.

You'll translate LLMs, agents, and RAG pipelines into polished product experiences—streaming chat, agentic workflows, intelligent summaries, and guided automation—while partnering with Platform, Product, and Design to connect AI capability to