Senior Machine Learning Engineer (United States)

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.

Responsibilitie s

• 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

Requiremen ts

• 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

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 , supported by profound expertise and ongoing investment in , , , and . Founded in 2006, Grid Dynamics is headquartered in Silicon Valley with offices across the Americas, Europe, and India.