Staff Machine Learning Engineer

Shipt is a retail tech company that connects people to reliable, high-quality delivery with a personal touch. As a Staff Machine Learning Engineer on Shipt's Personalization Platform team, you will drive key AI initiatives and collaborate with Data Scientists to design and deploy intelligent, personalized recommendation models.

Responsibilities:

• Drive key AI initiatives
• Collaborate with Data Scientists to design and deploy intelligent, personalized recommendation models
• Design, build and maintain scalable ML infrastructure and production systems
• Closely collaborate with Data Scientists to architect data pipelines for serving and training ML models
• Lead the team to build scalable, low-latency, and fault-tolerant systems to serve real-time recommendations at scale
• Develop robust APIs and services to deliver real-time or batch recommendations

Requirements:

• PhD/MS in Computer Science, Engineering, Mathematics or equivalent work experience
• 5+ years of experience in machine learning and backend software engineering
• Proficiency in at least one backend programming language (e.g., Go, Java) and Python
• Deep understanding of user modeling, embeddings, similarity search and ranking models
• Strong experience with serving architectures (e.g., REST/gRPC APIs, model servers)
• Experience with ML pipeline tools (e.g., MLflow, Kubeflow, Airflow)
• Strong grasp of distributed systems, microservices, and system design
• Experience with SQL and NoSQL databases and working with large-scale datasets
• Back-End Development
• Machine Learning (ML)
• Python (Programming Language)
• Bachelor's Degree or equivalent experience
• Exposure to experimentation platforms and online A/B testing