Principal Data Scientist

Walmart

. Applying machine learning ranking algorithms to maximize the relevance of search system and minimize risks.

Posted 4/9/2026full-timeSunnyvale • California • 🇺🇸 United StatesLead💰 $110,000 - $286,000 per yearWebsite

Tech Stack

Tools & technologies

HadoopJavaPythonPyTorchScalaSparkTensorflow

About the role

Key responsibilities & impact

Applying machine learning ranking algorithms to maximize the relevance of search system and minimize risks.
Building, validating, testing and deploying predictive models using machine learning techniques to explain or predict behavior and solve a variety of business and engineering problems.
Identifying, collecting, and exploring the right data used for predictive modeling and algorithm development.
Performing and leading developments using machine learning techniques in one or more of the key areas like natural language processing, computer vision, information retrieval, entity recognition, product classification, and recommender system.

Requirements

What you’ll need

At least 7 years of work and research experience in the machine learning field with a deep understanding of machine learning and interest in applying it at scale.
You have experience with data cleaning, preparation, and featurization and selection techniques.
You have experience working with large data sets to solve problems.
You have experience with deep learning library like TensorFlow and/or PyTorch is highly desirable as well as experience with Spark and/or Hadoop.
You have/had hands-on experience with Python, Java or Scala and the ability to write reusable and efficient code to automate machine learning pipeline and data processes.
Ph.D. or Master’s degree in Computer Science, Statistics, Artificial Intelligence, Operations Research, Mathematics, Electrical Engineering, Computational Linguistics or related fields.

Benefits

Comp & perks

Beyond competitive pay, you can receive incentive awards for your performance.
Other great benefits include a 401(k) match, stock purchase plan, paid maternity and parental leave, PTO, multiple health plans, flexible working hours, and more.
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 (including sick leave), parental leave, family care leave, bereavement, jury duty, and voting.
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 Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities.

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

machine learningpredictive modelingdata cleaningdata preparationfeature selectiondeep learningalgorithm developmentnatural language processingcomputer visionrecommender systems

Certifications

Ph.D.Master’s degree