Our client is transforming grocery shopping by giving people time back - shoppers handpick fresh groceries and household essentials and deliver them in as little as one hour. We're hiring a Senior Machine Learning Engineer for our Personalization Platform team within the Membership organization. You'll partner closely with Data Scientists to design, deploy, and scale personalized recommendation and ranking models that drive engagement across the Shipt marketplace. This is a hands-on senior technical role spanning architecture, ML infrastructure, and production systems - from training and serving pipelines to CI/CD and monitoring. If you have an AI-first mindset and enjoy building platforms that make other engineers and scientists more effective, this is your team.
Essential functions
Partner with Data Scientists, Product Managers, and peer engineering teams to build and ship AI product features
Design, build, and maintain scalable ML infrastructure and production systems, including batch and streaming data pipelines for model training and serving
Deploy, serve, and monitor models in public cloud environments via microservices that meet throughput and latency requirements
Define best practices for AI product development: pipeline automation, model versioning, deployment, and scalable serving
Design and manage CI/CD pipelines for machine learning deployments
Integrate models into search and ranking systems (e.g., Elasticsearch first-stage retrieval and downstream ranking services)
Own accountability for the implementation of AI product features end to end
Use modern monitoring and telemetry tooling to identify, assess, and prioritize platform issues
Support experimentation and A/B testing infrastructure, offline evaluation harnesses, and search-quality reporting
Stay current on emerging AI/ML technology and bring the best of it to the team
Qualifications
MS or PhD in Computer Science, Engineering, Mathematics, or equivalent practical experience.
5+ years in machine learning and backend software engineering.
Strong proficiency in Python, plus at least one other backend language (Go or Java preferred).
Deep understanding of user modeling, embeddings, similarity search, and ranking models.
Strong experience with serving architectures (REST/gRPC APIs, model servers) and low-latency inference.
Hands-on experience with modern AI development tooling - MCP servers/clients, PydanticAI, LangGraph, LiteLLM, or similar.
Experience with ML pipeline and experiment-tracking tools (MLflow, Kubeflow, Airflow, Weights & Biases, or similar).
Solid grasp of distributed systems, microservices, and system design.
Cloud experience (AWS, GCP, or Azure) and containerized development (Docker, Kubernetes).
Experience with SQL and NoSQL databases and large-scale datasets.
Familiarity with the full data science workflow - data ingestion, feature engineering, experimentation, deployment, and monitoring.
Exposure to experimentation platforms and online A/B testing.
Strong written and verbal communication; able to explain ML results and applications to scientists, engineers, and business partners
We offer
Opportunity to work on cutting-edge projects
Work with a highly motivated and dedicated team
Competitive salary
Flexible schedule
Benefits package - medical insurance, vision, dental, etc.
Corporate social events
Professional development opportunities
Well-equipped office
Please note that all onboardings must occur in person and you may be asked to travel to attend.
About us
Grid Dynamics (NASDAQ: GDYN) is a leading provider of technology consulting, platform and product engineering, AI, and advanced analytics services. Fusing technical vision with business acumen, we solve the most pressing technical challenges and enable positive business outcomes for enterprise companies undergoing business transformation. A key differentiator for Grid Dynamics is our 8 years of experience and leadership in enterprise AI, supported by profound expertise and ongoing investment in data, analytics, cloud & DevOps, application modernization and customer experience. Founded in 2006, Grid Dynamics is headquartered in Silicon Valley with offices across the Americas, Europe, and India.
| Location | CA |