Staff Data Scientist — Personalization & Vision at Scale

Walmart Inc.; Sunnyvale (CA); Hybrid; USD 225,000 - 286,000; Full time

Benefits offered by this job

Health benefits: medical, vision, andd

401(k) and stock purchase plan

Paid time off including sick leave and

Education assistance and lifeinsurance

Job summary

Walmart Inc. is seeking a Staff Data Scientist to design, develop, and productionize large-scale personalization and computer-vision systems at Sunnyvale, CA. You will lead data pipelines, collaborate with Eng to deploy models, and shape the ML roadmap with Product teams.

The role requires a Master’s or equivalent and 2 years (or a Bachelor’s with 4 years) of analytics experience; strong Python/SQL and CV/ML framework expertise are essential.

Qualifications

• Master’s degree or equivalent in Statistics, Economics, Analytics, Mathematics, Computer Science, Information Technology, Engineering (any), or related field and 2 years of experience in an analytics related field.
• Bachelor’s degree or equivalent in Statistics, Economics, Analytics, Mathematics, Computer Science, Information Technology, Engineering (any), or related field and 4 years of experience in an analytics related field.

Responsibilities

• Design, develop, and productionize large-scale personalization and computer-vision systems (candidate generation, ranking, visual similarity, image-based recommendations).
• Design, implement, and support deployment of large-scale data pipelines; collaborate with engineering to integrate models into serving systems.
• Contribute to discovery and formulation of the ML/DL roadmap for personalization and lead selected phases of the product lifecycle in partnership with Product and Engineering.
• Collaborate on online‑inference deployments, analyze pre/post‑deployment metrics, and recommend performance/tuning changes.
• Adapt pretrained vision and multimodal models via transfer learning for retrieval, ranking, and relevance; manage embeddings/feature stores.
• Develop generative‑modeling workflows for automated image/text asset creation (e.g., diffusion/GAN/VAE‑style or equivalent) with quality/brand checks and measured impact.
• Contribute input to metric definitions and experiment design; collaborate on A/B and offline relevance evaluation; present readouts with recommendations.
• Contribute to architecture and road‑mapping via design discussions and documentation; contribute to invention disclosures/patents.

Skills

Python

SQL

Unix Shell

Git

NumPy

Pandas

Matplotlib

Spark/PySpark

Scalding

Computer Vision

PyTorch

TensorFlow

CNN embeddings

Multimodal models

Transfer learning

Text embeddings

TF-IDF

Word2Vec

Doc2Vec

RNN/LSTM

ML algorithms

Classification

Clustering

SVM

Recommender systems

Matrix factorization

Collaborative filtering

Embedding retrieval

A/B testing

Offline relevance metrics

Production ML systems

Batch/stream processing

Evaluation of deployments

Generative modeling

GANs

VAEs

Diffusion

Foundation-model adaptation

Education

Master’s or equivalent in Statistics, Economics, Analytics, Mathematics, CS, IT, Engineering or related field

Bachelor’s or equivalent in Statistics, Economics, Analytics, Mathematics, CS, IT, Engineering or related field

Tools

Spark/PySpark

Scalding

Job description

Walmart Inc. is seeking a Staff Data Scientist to design, develop, and productionize large-scale personalization and computer-vision systems at Sunnyvale, CA. You will lead data pipelines, collaborate with Eng to deploy models, and shape the ML roadmap with Product teams.

The role requires a Master’s or equivalent and 2 years (or a Bachelor’s with 4 years) of analytics experience; strong Python/SQL and CV/ML framework expertise are essential.