Drata

Senior AI Product Engineer

Drata
Apply
4 hours ago

Base Salary

$150k - $230k/yr

Responsibilities

  • Build customer-facing AI-powered product features end-to-end using React, TypeScript, Node.js, and Python.
  • Translate LLM, RAG, agentic workflow, and structured reasoning capabilities into intuitive product interactions.
  • Partner with AI Engineers to define API contracts and output schemas.
  • Build feedback loops and observability for AI output quality, latency, engagement, task completion, and drop-off.
  • Implement streaming and real-time interfaces with graceful degradation and human-in-the-loop interaction patterns.
  • Establish reusable streaming hooks, feedback components, and AI state-management patterns.
  • Mentor engineers and participate in technical and cross-functional design reviews.

Requirements

  • 5+ years of software engineering experience with a record of shipping full-stack production features.
  • 2+ years of experience building AI-powered product features.
  • Strong React and TypeScript proficiency, including responsive interfaces handling asynchronous, streaming, and real-time data.
  • Solid Python and Node.js skills and experience designing and consuming APIs for LLM providers, orchestration frameworks, or AI services.
  • Hands-on experience with LLM APIs such as OpenAI and Anthropic, prompt engineering, context management, RAG, quality/evaluation systems, and structured outputs.
  • Familiarity with agent frameworks such as LangGraph, LangChain, or custom runtimes.
  • Experience implementing SSE, WebSockets, or chunked HTTP streaming and interfaces for progressive or uncertain outputs.
  • Strong product and user-experience judgment and experience collaborating with AI/ML engineers, product managers, and designers.
  • Preferred experience in compliance, security, GRC, document-heavy enterprise domains, MCP, RAG system design, or AI feedback and annotation tooling.

Benefits

  • Stock equity and a comprehensive total rewards package.
  • Up to 100% employer-paid medical, dental, and vision premiums for employees and dependents, plus wellness benefits and healthcare concierge services.
  • 401(k), company-paid life and disability insurance, tax-advantaged spending accounts, and discounted voluntary benefits.
  • Paid parental leave after six months, plus Kindbody fertility and family-building benefits and leave specialists.
  • Annual professional and personal development stipends and internal learning opportunities.
  • Flexible vacation policy, paid holidays, and other time-off benefits.
Drata

About Drata

5,001-10,000 employees

Drata builds a trust management platform that automates security compliance, risk, and audit readiness across frameworks like SOC 2, ISO 27001, HIPAA, and PCI DSS. Delivered as SaaS, it connects to cloud and business systems to monitor controls, manage GRC workflows, and streamline security reviews for thousands of organizations. Founded in 2020 and privately held, Drata is headquartered in San Francisco, California.

Contact me