Skip to content
DrataDrataSan Francisco, CA

Staff AI Engineer

Staff AI Engineer designs and owns end-to-end production AI systems including LLM pipelines, RAG, and agentic workflows for compliance platforms. Requires 10+ years software engineering with 3+ years in ML/AI, strong Python, and production AI experience.

201k – 272k/yr
Remote10+ YOEML Engineering

About the role

Responsibilities

Shape the Architecture

  • Design and own production AI systems end-to-end (LLM pipelines, RAG, reranking, vector stores, orchestration).
  • Make thoughtful build/don't-build decisions based on real data.
  • Evolve our AI stack over time, from model infrastructure to workflow orchestration to evaluation tooling.

Raise the Quality Bar

  • Design evaluation systems that measure retrieval quality, reasoning accuracy, end-to-end performance.
  • Build tooling that helps the team iterate confidently and catch regressions early.
  • Define how we measure success across the platform.

Investigate Deeply & Decide with Evidence

  • Analyze production outputs to identify failure patterns and root causes.
  • Turn complex findings into clear technical recommendations.
  • Know when to push forward and when to pause based on data.

Lead Across Teams

  • Be the go-to technical voice for AI architecture decisions.
  • Influence standards for how LLM systems are built, tested, deployed, and monitored.
  • Mentor senior engineers through design reviews and hands-on collaboration.
  • Partner with product and compliance teams to translate domain complexity into clean technical solutions.

Build Responsible, Production-Ready AI

  • Ship systems optimized for latency, cost, reliability, and auditability.
  • Embed safety guardrails, confidence thresholds, and human-in-the-loop workflows.
  • Ensure outputs are traceable and explainable.

Requirements

  • 10+ years of software engineering experience, including 3+ years working directly on ML/AI systems.
  • Real ownership of production LLM systems.
  • Deep experience with RAG, embeddings, reranking, vector databases (Pinecone, FAISS, Chroma, etc.), and agentic workflows.
  • Experience designing evaluation frameworks and using quantitative analysis to improve system performance.
  • Strong Python skills (TypeScript is a plus).
  • A track record of making architectural decisions that shape team direction.
  • Experience operating AI systems in production - observability, reliability, cost tradeoffs.
  • The ability to break down ambiguous, high-stakes problems into structured investigations.
  • Clear communication skills and comfort working cross-functionally.

Nice to Have

  • Experience in compliance, security, or other regulated domains.
  • Familiarity with enterprise data platforms or Snowflake-based analytics.
  • Experience with orchestration systems like Temporal or Airflow.
  • Experience building or using LLM evaluation platforms (e.g., Braintrust).
  • Contributions to technical communities or published work.

Compensation

  • Competitive base salary, benefits, and stock (RSUs).
  • Salary range: $200,700 - $271,500 (varies by tier: Tier 1 $220,800-$298,800; Tier 2 $198,700-$268,900; Tier 3 $176,600+).

Skills

LLMsRAGEmbeddingsRerankingVector DatabasesPineconeFaissChromaAgentic WorkflowsPythonTypeScriptEvaluation FrameworksOrchestrationTemporalAirflow

Similar roles

ML Engineering jobs
Clear Street

Senior / Staff AI Platform Engineer

Clear StreetUnited States

Build and own the core AI platform powering an AI-native trading copilot. Develop high-performance Rust backend for streaming, tool execution, and safe trading actions; design robust APIs with observability and security. Requires 8+ years systems programming experience.

200k – 350k/yr
Remote8+ YOEML Engineering
Clear Street

Senior / Staff AI Model Engineer

Clear StreetUnited States

Own reliability and quality for an AI copilot in a trading platform. Design evaluation systems, benchmarks, quality gates, model improvement loops, and AI monitoring for correctness, safety, and performance in market analysis and trading workflows. Requires 8+ years production software experience and strong ML eval expertise.

200k – 350k/yr
Remote8+ YOEML Engineering
Decagon

Staff Software Engineer, Agents

DecagonSan Francisco, CA

Build and own end-to-end AI agents for enterprise customers, integrating latest text/voice models and iterating based on real-world usage. Requires 8+ years of software engineering experience with Python and TypeScript.

200k – 400k/yr
On-site8+ YOEML Engineering
Nuance Labs

Member of Technical Staff - Research Fellow

Nuance LabsSeattle, WA

3-month research fellowship for early-career researchers working on frontier Multimodal LLMs, generative modeling, and real-time audiovisual AI. Own a research problem in pretraining, post-training, RL, evaluation, or multimodal modeling. Strong PyTorch and first-author tier-1 paper required.

200k – 250k/yr
On-siteML Engineering
Nuance Labs

Member of Technical Staff — Model Optimization and Inference

Nuance LabsSeattle, WA

Early-career engineer optimizing inference for real-time multimodal AI avatars. Focus on KV cache strategies, serving frameworks, quantization, and latency reduction for LLMs and diffusion models.

200k – 300k/yr
On-siteEntry levelML Engineering