Design and build end-to-end ML systems for Search relevance, Recommendation models, and Personalization engines.
Develop Agentic AI frameworks that enable autonomous, goal-driven personalization and next-best-action modeling.
Build and deploy scalable, low-latency ML services leveraging state-of-the-art infrastructure (Kubernetes, Databricks, Vertex AI, etc.).
Apply NLP, large language models (LLMs), and representation learning to understand intent, context, and user preferences.
Partner with Data Science, Product, and Engineering teams to drive experimentation, A/B testing, and continuous model improvement.
Contribute to model observability, governance, and explainability to ensure responsible AI practices.
Mentor junior engineers and influence technical direction across the Search & Personalization org.

5+ years of hands-on experience in machine learning engineering, preferably in Search, Recommender Systems, or Personalization.
Deep understanding of NLP, LLMs, and deep learning architectures (Transformers, BERT, T5, etc.).
Experience with Agentic or multi-agent AI systems — building or integrating AI agents that plan, reason, or act autonomously.
Strong programming skills in Python, Java, or Scala, with proficiency in TensorFlow, PyTorch, or JAX.
Experience deploying models at scale using cloud technologies (GCP, Azure, or AWS) and microservices frameworks.
Strong grasp of retrieval, ranking, embeddings, and personalization algorithms.
Excellent communication skills and a passion for building intelligent systems that impact millions.

Health benefits include medical, vision and dental coverage
Financial benefits include 401(k), stock purchase and company-paid life insurance
Paid time off benefits include PTO, parental leave, family care leave, bereavement, jury duty, and voting. You will also receive PTO and/or PPTO that can be used for vacation, sick leave, holidays, or other purposes. The amount you receive depends on your job classification and length of employment. It will meet or exceed the requirements of paid sick leave laws, where applicable.
Other benefits include short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement, and more.
Live Better U is a company paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities. Programs range from high school completion to bachelor's degrees, including English Language Learning and short-form certificates. Tuition, books, and fees are completely paid for by Walmart. Eligibility requirements apply to some benefits and may depend on your job classification and length of employment.

Senior Data Scientist

Key skills