Staff ML Engineer

Staff · Full Time · Remote

Palo Alto, CA · RemoteUSD 205k – 330k1mo ago
Apply for this role

Opens Docker's application page

Role

What you'll do.

Docker is seeking a Staff ML Engineer to join their Intelligence team, focusing on building cutting-edge ML systems for governance and security capabilities. The ideal candidate will be a hands-on technical leader who can design, implement, and ship advanced machine learning solutions that enhance Docker's platform for trusted autonomous workflows.

Responsibilities

  • ML System Design: Design and implement machine learning systems for governance and security capabilities
  • Infrastructure Development: Build comprehensive ML infrastructure including data pipelines, feature stores, and model serving systems
  • Technical Leadership: Set technical direction for ML work, define architecture, and establish evaluation methodologies
  • Model Lifecycle Management: Develop and maintain robust processes for model training, evaluation, and deployment
  • Team Growth: Contribute to team recruitment, mentorship, and organizational scaling

Qualifications

What we look for.

Technical

  • ML Expertise

    5+ years of deep applied ML/AI expertise with production system experience

  • Software Engineering

    8+ years of professional software engineering experience in backend or infrastructure roles

Education

  • Computer Science Degree

    Bachelor's degree in Computer Science, Engineering, or related field preferred

Experience

  • Production ML

    Proven track record of building and owning end-to-end ML systems

  • LLM Systems

    Experience with Large Language Model evaluation, prompt engineering, and production deployment

Skills

Required

  • Machine Learning

    Deep expertise in designing, training, and deploying production ML systems

  • AI Safety

    Experience with fraud detection, security, and trust domains

  • Large Language Models

    Proficiency in LLM-based systems, prompt engineering, and evaluation

  • Backend Engineering

    Strong software engineering skills with infrastructure and platform experience

  • Data Pipelines

    Ability to build and manage ML data infrastructure, feature stores, and model serving

Preferred

  • Agent Frameworks

    Nice to have

    Familiarity with autonomous agent ecosystems and multi-agent systems

  • Adversarial ML

    Nice to have

    Experience with detecting and mitigating adversarial attacks in ML systems

  • AI Governance

    Nice to have

    Understanding of policy, identity, and audit mechanisms for AI systems

Tech stack

Languages

PythonGo

Frameworks

PyTorchTensorFlowHugging Face

Databases

PostgreSQLRedis

Tools

KubernetesDockerMLflow

Other

CI/CDCloud Platforms

Compensation

Pay and benefits.

Base·USD 205,000 – 330,000

Equity·Stock options

Benefits

  • Flexible Work

    Remote-first culture with work flexibility

  • Parental Leave

    16 weeks of paid parental leave after 6 months of employment

  • Technology Stipend

    $100 monthly technology allowance

  • Professional Development

    Training stipend for conferences, courses, and classes

  • Equity

    Stock options to share in company success

  • Home Office Setup

    Support for comfortable home office environment

  • Time Off

    Quarterly Whaleness Days and end-of-year Whaleness break

Process

Interview steps.

  1. 01

    Initial Screening

    HR phone screen to assess basic qualifications and role fit

  2. 02

    Technical Phone Interview

    Detailed discussion of ML and software engineering background

  3. 03

    ML Systems Design Challenge

    In-depth technical interview focusing on ML architecture and problem-solving

  4. 04

    Onsite/Virtual Panel

    Multiple interviews with team members covering technical skills, system design, and cultural fit

  5. 05

    Final Leadership Interview

    Discussion with senior leadership about team vision and candidate's potential impact

Full posting

Original listing.

Docker has been one of the most loved brands in developer tooling, trusted by more than 20 million monthly users and over 20 billion container image pulls. From solo founders to the world's largest companies, developers rely on Docker to build, share, and run their applications across our suite of products including Docker Desktop, Docker Hub, and Docker Scout.

We are a globally distributed, remote-first team building the tools that define how software gets built and delivered. As AI agents redefine software development, Docker is at the center of that shift, providing the sandboxed environments, verified images, and secure infrastructure that make autonomous workflows trustworthy by default.

Docker's long-term vision is to become the runtime for trusted autonomy. As agents become more capable and autonomous, governance, policy, identity, and audit become foundational.

The Intelligence team builds intelligence-driven product capabilities that make software and agent execution on Docker safer, more effective, more trustworthy, and more efficient. Because Docker sits at the intersection of models, tools, software, identities, credentials, networks, and execution, we have visibility into behavior and context few other platforms can see, and we think that visibility is the foundation for a new layer of value across the platform.

About the role

We're hiring a Staff ML Engineer as one of the founding engineers on Intelligence Org. You'll work directly with the team's first engineers and manager to figure out what to build, how to build it, and how it fits into the broader Docker platform. This is a hands-on builder role with staff-level scope: you'll shape technical direction, ship the first versions of intelligence capabilities into customer hands, and grow the foundations (data, evaluation, infrastructure) the team will rely on as it scales.

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.

  • This role may require participation in a 24/7 on-call rotation for the Agentic Platform; carry genuine pager responsibility for the services you build and operate

Qualifications

  • 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.

  • You're 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.

Docker considers visa sponsorship on a case-by-case basis based on business needs.

Perks

  • Freedom & flexibility; fit your work around your life

  • Designated quarterly Whaleness Days plus end of year Whaleness break

  • Home office setup; we want you comfortable while you work

  • 16 weeks of paid Parental leave (after 6 months of employment)

  • Technology stipend equivalent to $100 USD net/month

  • PTO plan that encourages you to take time to do the things you enjoy

  • Training stipend for conferences, courses and classes

  • Equity; we are a growing start-up and want all employees to have a share in the success of the company

  • Docker Swag

  • Medical benefits, retirement and holidays vary by country

  • Remote-first culture, with offices in Seattle and Paris

Docker embraces diversity and equal opportunity. We are committed to building a team that represents a variety of backgrounds, perspectives, and skills. The more inclusive we are, the better our company will be.

#LI-REMOTE

Redirects to Docker's application page.

Other roles

More at Docker.

View all 24 roles