Staff ML Engineer: Real-Time Personalization & Scalable AI

Shipt; San Francisco (CA); Hybrid; USD 137,000 - 239,000; Full time

Job summary

Shipt is seeking a Staff Machine Learning Engineer on the Personalization Platform team to drive AI initiatives and collaborate with Data Scientists on personalized recommendations that boost user engagement.

You will architect and implement scalable ML infrastructure, design data pipelines for training and serving models, and lead the team to deliver low-latency, real-time or batch recommendations with robust APIs and services.

Qualifications

• PhD or MS in Computer Science, Engineering, Mathematics or equivalent real-world experience.
• 5+ years in machine learning and backend software engineering.
• Proficiency in Python and at least one backend language (Go/Java).
• Experience with user modeling, embeddings, similarity search and ranking models.
• Strong understanding of serving architectures and model servers.
• Experience with ML pipeline tools (MLflow, Kubeflow, Airflow).
• Knowledge of distributed systems, microservices, and system design.
• Experience with SQL and NoSQL on large-scale data.
• Exposure to experimentation platforms and online A/B testing.

Responsibilities

• Design, implement and deploy intelligent, personalized recommendation models to drive user engagement.
• Build and maintain scalable ML infrastructure and production systems.
• Collaborate with Data Scientists to architect data pipelines for training and serving ML models.
• Lead the team to deliver real-time or batch recommendations with low latency and fault tolerance.
• Develop robust APIs and services to support AI-driven features.

Skills

Backend development

Python

Go

Java

Distributed systems

REST/gRPC APIs

ML infrastructure

Education

PhD/MS in Computer Science, Engineering, Mathematics or equivalent

Tools

MLflow

Kubeflow

Airflow

Job description

Shipt is seeking a Staff Machine Learning Engineer on the Personalization Platform team to drive AI initiatives and collaborate with Data Scientists on personalized recommendations that boost user engagement.

You will architect and implement scalable ML infrastructure, design data pipelines for training and serving models, and lead the team to deliver low-latency, real-time or batch recommendations with robust APIs and services.