Staff Data Scientist

Walmart

Confirmed live 2 days ago High trust

Quick summary

Work type
On-site
Location
Sunnyvale, CA
Salary
$143,000–$286,000 / yr
Posted
60 days ago
Freshness
Confirmed live 2 days ago

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $176k
This role $214k
$96k most similar roles pay here $306k

This role pays more than 77% of similar roles. Most pay $136,875–$214,500 — the shaded band above. At the midpoint, this role pays about $214k versus about $176k for comparable roles.

Based on 240 similar postings.

Employer

About Walmart

Walmart Inc. is the world''s largest retailer by revenue, operating a chain of hypermarkets, discount department stores, and grocery stores, as well as a growing e-commerce presence through Walmart.com. Industry: General Merchandise & Grocery Retail

Walmart currently has 184 open roles on FindRole.

Listed pay typically runs $117,000–$234,000 across 180 roles with salary data.

Most-posted roles

View all roles at Walmart

At a glance

TL;DR · Staff Data Scientist

As a (USA) Staff, Data Scientist on the Spark Driver Applied AI team, you will lead a group of applied scientists to develop machine learning and reinforcement learning models for driver search and matching. You will be responsible for designing, training, evaluating, deploying, and monitoring production models while collaborating with business, product, and engineering teams to align AI objectives with corporate goals. Your daily work involves building end-to-end pipelines that balance accuracy and computational efficiency to improve delivery reliability and operational efficiency. The role requires expertise in Python, Java, C, or C++, along with experience in Spark/Hive data systems and GCP or Azure cloud environments. You will apply advanced techniques in deep learning, GenAI, and LLMs to solve complex problems involving capacity planning, offer pricing, surge pricing, incentive strategy, and ETA prediction within a large-scale delivery platform.

What does a Data Scientist earn in California?

Median $208978 from 66 postings across 27 companies.

See salary data

What you'll do

  • Lead a team of applied scientists to develop innovative AI-driven solutions for driver search and matching.
  • Research and implement state-of-the-art machine learning, reinforcement learning, and deep learning techniques.
  • Design and optimize scalable machine learning pipelines for real-time processing capabilities.
  • Build and deploy end-to-end production models while balancing accuracy and computational efficiency.
  • Integrate advanced AI technologies like LLMs to improve platform efficiency and profitability.
  • Develop models for capacity planning, pricing strategies, ETA prediction, and trust safety.
  • Ensure AI objectives align with business goals through cross-functional collaboration with product and engineering teams.

What we're looking for

  • Ph.D. or equivalent advanced degree in Computer Science, Statistics, Operations Research, Economics, or a related field.
  • Option 1: Bachelor’s degree and 4 years of experience in an analytics-related field.
  • Option 2: Master’s degree and 2 years of experience in an analytics-related field.
  • Option 3: 6 years of experience in an analytics or related field.
  • Extensive experience in machine learning, reinforcement learning, GenAI, and LLMs.
  • 5+ years of experience deploying ML models in production, preferably in high-scale environments.
  • Strong programming skills in Python, Java, C, or C++.
  • Experience with large-scale data systems (Spark/Hive) and cloud computing (GCP/Azure).

More like this

Similar roles

Staff Data Scientist

Walmart

Bellevue, WA 99 days ago $132,000$264,000
Machine Learning Python Spark SQL Hive Scala R Deep Learning Gen AI scikit learn TensorFlow Statistics Big Data Optimization Models Econometrics
4+ yrs exp

Staff Data Scientist

Walmart

Sunnyvale, CA +1 65 days ago $143,000$286,000
Machine Learning Python SQL NoSQL Spark Scala R scikit learn TensorFlow Tableau PowerBI Matplotlib Plotly Predictive Analytics Statistical Modeling Data Visualization Optimization Models WCAG) 2.2 AA
4+ yrs exp

Staff Data Scientist

Walmart

Bentonville, AR 115 days ago $110,000$220,000
Machine Learning Python Spark Scala SQL Kafka Hadoop Hive Pig Map Reduce SRE Git TensorFlow scikit learn
4+ yrs exp

Senior Data Scientist

Walmart

Sunnyvale, CA 53 days ago $117,000$234,000
Python SQL Machine Learning LLM VLLM PyTorch DeepSpeed Horovod MLOps Spark scikit learn TensorFlow Data Visualization Distributed Training Microservices R Scala
3+ yrs exp

Staff Data Scientist

Toast, Inc.

Remote 65 days ago $170,000$272,000
Machine Learning Python SQL AWS SageMaker Athena Glue DynamoDB Bedrock PyTorch TensorFlow scikit-learn LLMs MLOps CI/CD A/B Testing Causal Inference Time Series Deep Learning
7+ yrs exp Remote

Principal Data Scientist

Walmart

Sunnyvale, CA +1 81 days ago $143,000$286,000
Machine Learning Python Py Spark R Scala Google Cloud Platform Vertex AI Kubeflow Hadoop Hive Map Reduce CUDA TensorFlow scikit learn Gen-AI NLP Vector Databases ML Ops
10+ yrs exp