Drata

Senior AI Product Engineer 2, Audit

Drata
Apply
4 months ago

Base Salary

$192k - $260k/yr

Responsibilities

  • Build end-to-end AI-powered product features spanning React/TypeScript UI and Node.js/Python backend services.
  • Translate RAG pipelines, agentic workflows, and structured reasoning into intuitive compliance-practitioner experiences.
  • Partner with AI Engineers to define API contracts and output schemas.
  • Advocate for users, identify model-output breakdowns, and iterate on the product layer.
  • Build feedback loops that capture AI-output quality signals for the broader AI team.
  • Instrument AI features with product-level observability for latency, engagement, task completion, and drop-off.
  • Create integration and end-to-end tests validating behavior across the full stack.
  • Establish reusable streaming hooks, feedback components, and AI state-management patterns.
  • Mentor engineers newer to AI product development and participate in engineering and product design reviews.

Requirements

  • 7+ years of software engineering experience shipping full-stack production features, including 2+ years working on AI-powered product features.
  • Strong proficiency in React and TypeScript for responsive, interactive interfaces handling asynchronous, streaming, and real-time data.
  • Solid Python and Node.js skills and experience designing and consuming APIs integrating with LLM providers, orchestration frameworks, or AI services.
  • Hands-on experience with LLM APIs such as OpenAI and Anthropic, prompt engineering, context management, RAG, quality and evaluations, and structured outputs.
  • Familiarity with agent frameworks such as LangGraph, LangChain, or custom runtimes and multi-step agentic workflows.
  • Experience implementing streaming responses with SSE, WebSockets, or chunked HTTP and building interfaces for progressive and uncertain outputs.
  • Strong product sense and experience collaborating with AI/ML engineers, product managers, and designers in cross-functional teams.
  • Bonus: experience in compliance, security, GRC, or document-heavy enterprise domains.
  • Bonus: experience with MCP or similar AI-agent tool-use interfaces.
  • Bonus: exposure to RAG system design and debugging retrieval-quality issues from the product layer.
  • Bonus: experience building feedback and annotation tooling for AI output quality.

Benefits

  • Hybrid work model in the Bay Area with in-office collaboration Tuesday through Thursday and flexible Mondays and Fridays.
  • Stock equity, typically in the form of Restricted Stock Units (RSUs).
  • Up to 100% employer-paid medical, dental, and vision premiums for employees and dependents.
  • 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 of employment, plus Kindbody fertility and family-building benefits and leave specialists.
  • Annual professional and personal development stipends and internal learning opportunities.
  • Flexible vacation policy and paid holidays.

Categories

AI ApplicationsFull StackProduct Engineering
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