Principal Data Scientist: Associate AI experience

Walmart

Confirmed live yesterday High trust

Quick summary

Work type
On-site
Location
Bentonville, ARSunnyvale, CABellevue, WA
Salary
$110,000–$220,000 / yr
Posted
171 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $180k
This role $165k
$96k $239k
below market most similar roles pay here above market

This role pays less than 54% of similar roles. Most pay $145,150–$214,500 — the blue band above. At the midpoint, this role pays about $165k versus about $180k 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 719 open roles on FindRole.

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

Most-posted roles

View all roles at Walmart

At a glance

TL;DR · Principal Data Scientist: Associate AI experience

JOB TITLE: Principal Data Scientist: Associate AI experience The Principal Data Scientist joins the Associate AI Experiences team to provide technical leadership and long-term vision for next-generation AI systems. This role involves architecting end-to-end ML systems, including data pipelines, feature stores, and production deployment. You will build and deploy production agentic systems, autonomous AI agents, and multi-agent orchestration systems to solve high-impact business problems across stores and fulfillment centers. Key responsibilities include designing agentic evaluation frameworks, deploying NLP pipelines, and mentoring senior talent. You will utilize technologies including PyTorch, LangChain, Pydantic AI, RAG patterns, and FAISS for vector search. Required skills include expertise in XGBoost, CatBoost, LightGBM, and deep learning architectures like BERT and SBERT. You will also leverage AWS SageMaker and GCP Vertex AI to deliver scalable, production-grade AI solutions.

What does a Data Scientist earn in California?

Median $214500 from 134 postings across 32 companies.

See salary data

What you'll do

  • Architect and deploy end-to-end ML systems including data pipelines, feature stores, and production serving layers.
  • Build and ship autonomous AI agents using LangChain, Pydantic AI, and RAG patterns.
  • Design agentic evaluation frameworks to benchmark performance across task completion and multi-step reasoning.
  • Deploy enterprise-scale NLP pipelines using BERT/SBERT embeddings and vector search.
  • Lead high-impact projects from initial problem framing and prototyping through to production rollout.
  • Mentor senior scientists and engineers on advanced modeling techniques and experimentation design.
  • Establish best practices for model validation, experimentation, and safe deployment of AI systems.
  • Present technical findings and quantified business impacts to senior leadership.

What we're looking for

  • Master's degree in a quantitative field and 3 years of experience in an analytics related field, or a Bachelor's degree and 5 years of experience.
  • 7 years of experience in an analytics or related field is also accepted as a minimum qualification.
  • Deep expertise in ML algorithms (XGBoost, CatBoost, LightGBM, Random Forest, AutoML) and deep learning (PyTorch, transformer architectures).
  • Production experience with NLP (BERT, SBERT, FAISS, RAG), computer vision (YOLO, CLIP), and time series forecasting (ARIMA, Prophet).
  • Hands-on experience building and orchestrating multiple AI agents with LangChain, RAG, tool integration, and memory management.
  • Multi-cloud ML platform proficiency in AWS SageMaker and GCP (Vertex AI, BigQuery, BigQuery ML).
  • Strong Python skills (Pandas, NumPy, Scikit-learn) with containerization (Docker) and MLOps practices.
  • PhD in a related field, publications, or active peer review experience (preferred).

More like this

Similar roles

Distinguished Data Scientist, Associate AI Experience

Walmart

Bentonville, AR +2 171 days ago $130,000–$260,000
Agentic AI Python LangChain RAG PyTorch AWS SageMaker GCP (Vertex AI BigQuery MLOps Docker Scikit-learn XGBoost CatBoost LightGBM BERT FAISS NLP Computer Vision Time Series Forecasting Pydantic AI Spark TensorFlow Pandas NumPy
6+ yrs exp

Principal Data Scientist

Walmart

Bentonville, AR +2 55 days ago $110,000–$220,000
Agentic AI Python PyTorch LangChain RAG NLP Computer Vision Time Series Forecasting AWS SageMaker GCP (Vertex AI BigQuery Docker MLOps XGBoost CatBoost LightGBM Random Forest BERT SBERT FAISS Scikit-learn Pandas NumPy Spark Scala R TensorFlow
5+ yrs exp

Distinguished Data Scientist

Walmart

Bentonville, AR +2 58 days ago $130,000–$260,000
Agentic AI Python PyTorch LangChain RAG MLOps AWS SageMaker GCP (Vertex AI BigQuery FAISS XGBoost CatBoost LightGBM Random Forest BERT SBERT YOLO CLIP ARIMA Prophet Docker Scikit-learn Pandas NumPy Spark Scala R TensorFlow Time Series Forecasting Computer Vision NLP
8+ yrs exp

Principal Associate Data Scientist, Applied AI

Capital One Financial

New York, NY +1 16 days ago $161,800–$184,600
Generative AI Python LangChain AWS Vector Databases RAG Multi-Agent Systems Machine Learning Deep Learning NLP SQL Scala R Time Series Clustering Sentiment Analysis Statistical Modeling Data Analytics
5+ yrs exp

Senior Associate Data Scientist, Applied AI

Capital One Financial

New York, NY +1 16 days ago $135,600–$154,800
Generative AI Python PyTorch LangChain AWS Spark Vector Databases RAG Multi-Agent Systems Deep Learning NLP SQL Scala R Machine Learning Statistics Time Series Clustering Sentiment Analysis
2+ yrs exp

Principal Data Scientist, AI for Data and Assessment

Capital One Financial

McLean, VA +2 18 days ago $161,800–$184,600
Python SQL AWS Spark H2O Deep Learning Natural Language Processing Conda Machine Learning Statistical Modeling Clustering Classification Time Series Sentiment Analysis Big Data
5+ yrs exp