Skip to content
DockerDockerPalo Alto, CA

Staff ML Engineer

Founding Staff ML Engineer building production ML systems for governance, security, and agentic platform capabilities at Docker. Owns architecture, data pipelines, evaluation, and model lifecycle while mentoring the growing team.

205k – 330k
Remote8+ YOEML Engineering

About the role

Responsibilities

  • Design, train, evaluate, and ship ML systems that power governance and security capabilities, starting with problems like prompt injection detection, behavioral anomaly detection, trust scoring, and policy recommendations.
  • Build the supporting infrastructure: data pipelines, feature stores, model serving, evaluation harnesses, and the feedback loops that make iteration fast.
  • Make pragmatic build-vs-buy calls. Use frontier models, off-the-shelf tooling, and managed services to move quickly; invest in custom systems where they create durable advantage.
  • Set technical direction for the team's ML work. Own the architecture, evaluation methodology, model lifecycle, and the bar for shipping.
  • Help recruit, mentor, and shape the team as it grows.
  • Participate in a 24/7 on-call rotation for the Agentic Platform.

Requirements

  • 5+ years of deep applied ML/AI expertise with a track record of shipping production systems.
  • Experience in fraud, abuse, safety, security, or trust domains, where adversarial dynamics, imbalanced data, and high-stakes decisions is valuable.
  • 8+ years of professional, hands-on, full-time software engineering experience in backend, infrastructure, or platform engineering.
  • Bachelor's degree in Computer Science, Engineering, or a related field, or equivalent practical experience.
  • You've built and owned the systems around ML models, i.e. data pipelines, serving, evaluation, monitoring etc. and have shipped customer-facing products end to end.
  • You use modern AI tools fluently in your day-to-day work and have a sharp instinct for when frontier models can replace traditional ML, when they can't, and when to combine the two.
  • Experience with LLM-based systems in production - evaluation, prompt engineering, fine-tuning, retrieval, guardrails, agent frameworks.
  • Familiarity with the agent / MCP ecosystem.
  • Energized by an early-stage effort where the roadmap is being written as the work happens, and you make crisp decisions with incomplete information.
  • Collaborative and low-ego. You work well across teams, write clearly, and bring others along.

Skills

Machine LearningLLMsPrompt EngineeringFine-TuningRetrievalGuardrailsAgent FrameworksData PipelinesModel ServingEvaluationBackend EngineeringInfrastructure

Similar roles

ML Engineering jobs
Hinge Health

Staff Machine Learning Scientist

Hinge HealthSan Francisco, CA

Own ML systems for send-time optimization, propensity modeling, and nudge decisions at consumer scale. Set experimentation standards and mentor a small ML team.

205k – 307k
Hybrid7+ YOEML Engineering
Airbnb

Staff Machine Learning Engineer, Listings and Host Tools Data and AI

AirbnbUnited States

Builds and productionizes ML models for host personalization products using NLP and computer vision. Requires 8+ years experience, MS/PhD, strong programming, and expertise in ML infrastructure and large-scale data pipelines.

204k – 255k
Remote8+ YOEML Engineering
Airbnb

Staff GenAI System Engineer, Automation Foundation

AirbnbUnited States

Leads development of Airbnb's conversational AI Automation Platform and agent provisioning systems. Drives backend optimization, prompt engineering, and LLM integrations with 9+ years experience in scalable architectures.

204k – 255k
Remote9+ YOEML Engineering
Rula

Staff AI Engineer - Applied AI (Remote)

RulaLos Angeles, CA

Leads applied AI investments, owning technical direction for high-impact AI products from design to production deployment. Requires 8+ years AI/ML experience, Python expertise, foundation models, LLM patterns, and evaluation systems.

207k – 243k
Remote8+ YOEML Engineering
Watershed

Staff Software Engineer, AI Platform

WatershedSan Francisco, CA

Technical leader building agent infrastructure, observability, evals, and guardrails for production AI systems at Watershed. Requires 6+ years backend/platform/AI engineering experience and production TypeScript systems.

202k – 255k
On-site6+ YOEML Engineering