Skip to main content
Back to jobs

Senior, Data Scientist

External
Walmart logoWalmart · Milpitas, CA
Full-timeOn-site9mo ago
A/B TestingDeep LearningEpicLLMsMachine LearningMLOps
Cover LetterConnect

Prepare for this interview

Elite

AI-generated questions, company research, and talking points tailored to this role


About the role

Our team works closely with our US stores and eCommerce business to better serve customers by empowering team members, stores, and merchants with technological innovation. From groceries and entertainment to sporting goods and crafts, Walmart U.S. offers an extensive selection that our customers value, whether they shop online at Walmart.com, through one of our mobile apps, or in-store. Focus areas include customers, stores and employees, in-store service, merchant tools, merchant data science, and search and personalization.

Responsibilities

  • Design, develop, and productionalize machine learning models for personalized recommendations, layout optimization, and content generation (visual and textual) using structured and unstructured data.
  • Lead the development of large-scale ML models leveraging deep learning (Transformers, multi-modal, LLMs), reinforcement learning, and personalization frameworks (e.g., agentic systems).
  • Collaborate cross-functionally with engineering, product, and design teams to integrate ML systems into customer-facing applications.
  • Build scalable data pipelines and feature stores with data engineers for real-time and batch personalization use cases.
  • Employ and experiment with pre-trained foundation models (e.g., BERT, RoBERTa, T5, CLIP) for customer representation and product understanding.
  • Apply best-in-class practices in model validation, monitoring, explainability (e.g., SHAP, counterfactuals), and lifecycle management.
  • Contribute to research and innovation in personalization by exploring novel architectures and publishing internally or externally where applicable.

Requirements

  • Have a MS/PhD in relevant technical degree (preferably Computer Science, Machine Learning, Operations Research, Applied Mathematics, Statistics, Engineering etc.) with 2-4 years of relevant work experience.
  • Proven experience developing and deploying ML models at scale in a production environment, especially for recommendation systems, ranking, or NLP tasks.
  • Proficiency in Python, SQL, and PySpark. Strong familiarity with ML libraries such as TensorFlow, PyTorch, Scikit-Learn, XGBoost.
  • Strong knowledge of ML evaluation and diagnostic techniques (A/B testing, ROC-AUC, Precision@K, RMSE, etc.).
  • Experience with foundation models and LLMs
  • Background in reinforcement learning or agent-based personalization frameworks.
  • Experience working with big data tools (Hive, Spark) and modern MLOps practices.
  • About Walmart Global Tech
  • Our hubs: Walmart is a global company with offices across the United States and around the world. Our global headquarters is in Bentonville, Arkansas, with primary hubs in the San Francisco Bay area and New York/New Jersey.

Benefits

Benefits: Beyond our great compensation package, you can receive incentive awards for your performance. Other great perks include 401(k) match, s

Additional Information

Position Summary... What you'll do... Walmart Global Tech is seeking a highly skilled and experienced Senior Data Scientist to join our Personalization and Recommendation team. This role will be instrumental in driving forward Walmart's personalization capabilities across multiple touchpoints - from product recommendations and layout optimization to GenAI-powered customer understanding. The ideal candidate combines deep ML expertise with large-scale production experience and thrives in fast-paced, impact-driven environments. As part of our team, you'll join a fast-paced and collaborative environment that blends research with real-world impact. You'll work closely with world-class engineers, product managers, and fellow scientists to push the boundaries of what personalization means in modern retail - crafting solutions that are fair, relevant, and truly customer-centric at a global scale.


Your Match

How well this role fits your profile.

Company Intel

What employees say

Worked at Walmart? Share your experience

Interested in this role?

Apply on the company's website.

Cover LetterConnect