Staff Data Scientist — E-commerce Search ML

Walmart Global Tech; Sunnyvale (CA); On-site; USD 143,000 - 286,000; Full time

Benefits offered by this job

401(k) match

Stock purchase plan

Paid maternity and parental leave

PTO

Flexible working hours

Job summary

Walmart Global Tech in Sunnyvale, CA, is seeking an experienced ML Engineer to accelerate the relevance of search and product discovery. Lead end-to-end ML projects, from data collection to deployment, focusing on scalable models for e‑commerce queries and categories.

The role requires deep learning expertise, handling vast data sets, and strong coding in Python/Java/Scala, with familiarity in TensorFlow or PyTorch and Spark/Hadoop experience.

Qualifications

• 5+ years of work/research experience in machine learning with ability to apply at scale.
• Experience with data cleaning, preparation, and feature selection techniques.
• Experience handling large data sets to solve problems.

Responsibilities

• Apply machine learning to maximize relevance of search systems and minimize risks.
• Build, validate, test, and deploy predictive models for business and engineering problems.
• Identify and explore data used for predictive modeling and algorithm development.
• Lead developments in NLP, computer vision, information retrieval, entity recognition, product classification, and recommendations.

Skills

Machine learning

Deep learning

Python

Natural language processing

Information retrieval

Data cleaning

Working with large data sets

Communications

Education

PhD or Master’s in CS/Statistics/AI/OR related fields

Tools

TensorFlow

PyTorch

Spark

Hadoop

Job description

Walmart Global Tech in Sunnyvale, CA, is seeking an experienced ML Engineer to accelerate the relevance of search and product discovery. Lead end-to-end ML projects, from data collection to deployment, focusing on scalable models for e‑commerce queries and categories.

The role requires deep learning expertise, handling vast data sets, and strong coding in Python/Java/Scala, with familiarity in TensorFlow or PyTorch and Spark/Hadoop experience.