Staff Machine Learning Engineer
Role details
Job location
Tech stack
Job description
As a Staff Machine Learning Engineer on Shipt's Personalization Platform team you will drive key AI initiatives. In this role, you'll collaborate with Data Scientists to design and deploy intelligent, personalized recommendation models to drive user engagement across Shipt. You will be a hands-on senior technical contributor in the Membership organization and your work will consist of contribution to architecture, design, and implementation to deliver high quality AI driven solutions in a rapidly evolving marketplace environment. In this role you will design, build and maintain scalable ML infrastructure and production systems as well as closely collaborate with Data Scientists to architect data pipelines for serving and training ML models for Shipt's Personalization Platform. You will also be responsible for leading the team to build scalable, low-latency, and fault-tolerant systems to serve real-time recommendations at scale and developing robust APIs and services to deliver real-time or batch recommendations (using Python, Go). This is a hands-on, cross-functional role ideal for someone who thrives at the intersection of Machine Learning, AI Engineering, and Business Impact.
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
-
Exposure to experimentation platforms and online A/B testing.
Pay Range: $137,400- $239,000 Please note that the salary range above is a guideline, and individual total compensation will vary based on factors such as qualifications, skill level, competencies, and work location.
Skills & Education
This list includes key skills used in this job but is not inclusive of all skills needed for the role. Please see any required education below.
Back-End Development, Machine Learning (ML), Python (Programming Language), Structured ProgrammingBachelor's Degree or equivalent experience | Required
Work Arrangement
To foster connection while offering continued flexibility, hybrid team members have the following in-office expectations:
Hybrid roles in Birmingham, AL typically work in-office at least 2 days per week, with core in-office days on Wednesdays and Thursdays.
Hybrid roles in Minneapolis, MN typically work in-office at least 1 day per week on either Tuesday, Wednesday, or Thursday.
Hybrid roles in San Francisco, CA typically work in-office at least 1 day per week between Monday and Thursday.
Certain roles may require in-office presence on a full-time basis. Please work with your recruiter to learn more about the classification of this role.
Benefits & conditions
Please inform your recruiting contact upon initial connection if you need a reasonable accommodation. If you need assistance filling out a job application, please complete this form.
Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.
Employees (and eligible family members) are covered by medical, dental, vision and more. Employees may enroll in our company's 401k plan. Employees will also be eligible to receive discretionary vacation for exempt team members, paid holidays throughout the calendar year and paid sick leave. Other compensation includes eligibility for an annual bonus and the potential for restricted stock units based on role.