Staff ML Platform Engineer: Architect AI Infra & Scale

Attentive

United States

On-site

USD 170,000 - 280,000

Full time

14 days+
Application generator

Turn this role into an interview — a resume and cover letter built around what this employer wants.

Get past ATS filters

Job summary

Attentive is seeking a Staff Software Engineer for its Machine Learning Platform team. The role involves architecting AI/ML infrastructure, influencing strategy, and mentoring team members. You will build foundational components for AI capabilities with a focus on performance and reliability.

This position offers a competitive salary range of $170,000 - $280,000 annually, plus equity and benefits. The ideal candidate will have 5+ years of MLOps experience and a deep understanding of ML tools.

Qualifications

  • 5+ years of experience specifically in ML Platform/MLOps.
  • Proven record of managing ML platform components.
  • Strong understanding of online and offline inference systems.

Responsibilities

  • Architect ML platform strategy spanning data pipelines and training infrastructure.
  • Provide technical leadership and mentorship in ML infrastructure discussions.
  • Build architectures that bridge platform capabilities with product needs.

Skills

ML Platform/MLOps
Python
Experience with Spark
Experience with Ray
Knowledge of MLFlow
Knowledge of Kubeflow
Model lifecycle systems

Tools

Argo
Metaflow

Job description

Attentive is seeking a Staff Software Engineer for its Machine Learning Platform team. The role involves architecting AI/ML infrastructure, influencing strategy, and mentoring team members. You will build foundational components for AI capabilities with a focus on performance and reliability.

This position offers a competitive salary range of $170,000 - $280,000 annually, plus equity and benefits. The ideal candidate will have 5+ years of MLOps experience and a deep understanding of ML tools.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Staff ML Platform Engineer - Scale AI Infra & Equity
Staff ML Platform Engineer - Scale AI Infra & Equity

Attentive • Wilmington (DE)

On-site
USD 170,000 - 280,000
Health & wellness benefits
Equity opportunities
Staff Engineer, ML/AI Platform
Staff Engineer, ML/AI Platform

Attentive • United States

On-site
USD 170,000 - 280,000
Staff ML Platform Engineer — Production AI Infra (Remote)
Staff ML Platform Engineer — Production AI Infra (Remote)

AppFolio, Inc • Santa Barbara (CA)

Hybrid
USD 200,000 - 250,000
Staff AI Infrastructure Engineer – ML Platform & Edge
Staff AI Infrastructure Engineer – ML Platform & Edge

Anduril Industries • Costa Mesa (CA)

On-site
USD 220,000 - 292,000
Staff AI Platform Engineer: Scalable ML Infra
Staff AI Platform Engineer: Scalable ML Infra

LinkedIn • California (MO)

Hybrid
USD 175,000 - 287,000
Staff ML Platform Engineer: Scalable AI Infra & Systems
Staff ML Platform Engineer: Scalable AI Infra & Systems

Stripe • Seattle (WA)

On-site
USD 224,000 - 336,000
Senior ML Platform Engineer — Scale AI Infrastructure
Senior ML Platform Engineer — Scale AI Infrastructure

Socket.dev • Denver (CO)

Hybrid
USD 160,000 - 240,000
Staff Machine Learning Platform Engineer — Scale ML Infra
Staff Machine Learning Platform Engineer — Scale ML Infra

Stripe • San Francisco (CA)

On-site
USD 150,000 - 200,000
Senior Staff ML Engineer - Scalable LLM Infra
Senior Staff ML Engineer - Scalable LLM Infra

Moveworks • Mountain View (CA), Northern (KY)

Hybrid
USD 190,000 - 280,000
Senior ML Platform Engineer – Real-Time Data & Features
Senior ML Platform Engineer – Real-Time Data & Features

Attentive • United States

On-site
USD 220,000 - 260,000
Competitive perks and benefits
Equity benefits