Principal Data Scientist: Associate AI experience

Walmart

Actively hiring
(Usa) Bentonville Global Tech Ar Bentonville Home Office, US Posted 35 days ago $110,000–$220,000 / year
View original post Log in to save

At a glance

AI generated

TL;DR

We are seeking a Principal Data Scientist to join the Associate AI Experiences team at Walmart, where you will provide technical leadership and vision for next-generation AI systems. This role involves architecting end-to-end ML/AI solutions from feature engineering through production deployment, building autonomous AI agents, and deploying NLP pipelines at enterprise scale using technologies like BERT/SBERT and FAISS. You will also mentor senior talent, influence roadmap prioritization, and present findings to leadership with clear business impact. Ideal candidates have deep expertise in ML algorithms, deep learning frameworks (PyTorch), production experience with NLP and computer vision tools, proficiency in multi-cloud platforms, and strong Python skills for MLOps practices. This role demands a high level of technical rigor combined with practical business acumen to drive measurable value at Walmart’s global scale.

Skills

Python Pandas NumPy Scikit-learn Docker AWS SageMaker GCP Vertex AI BigQuery BigQuery ML BERT SBERT FAISS RAG YOLO CLIP ARIMA Prophet XGBoost CatBoost LightGBM Random Forest AutoML PyTorch LangChain MLOps

What you'll do

  • Build and deploy end-to-end ML systems including data pipelines, feature stores, model training, serving layers, monitoring, and feedback loops.
  • Provide technical vision and lead advanced development in agentic AI, reinforcement learning, and simulation.
  • Architect large-scale AI systems and autonomous agents using LangChain, Pydantic AI, and RAG patterns.
  • Design evaluation frameworks to benchmark agent performance across various metrics at enterprise scale.
  • Mentor senior scientists and engineers on advanced modeling techniques and best practices for safe deployment of AI systems.

What we're looking for

  • Extensive hands-on experience in building and deploying end-to-end ML/AI systems.
  • Deep expertise in advanced machine learning algorithms, deep learning frameworks, and NLP technologies.
  • Hands-on experience with multi-cloud ML platforms (AWS SageMaker, GCP Vertex AI).
  • Proficiency in Python for data science tasks including containerization and MLOps practices.
  • Strong background in statistical analysis, experiment design, and causal inference.
  • Ability to mentor senior talent and influence long-term strategy for AI systems.

Market check

Salary context

This $110,000–$220,000 range sits above 53% of similar postings on FindRole.

Peer median band

$112,800–$225,000

Median floor and ceiling across peers.

Typical midpoint (25–75%)

$135,000–$205,800

Middle half of comparable postings.

Based on 240 comparable postings.

* 240 is the maximum number of comparable postings sampled.

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 495 open roles on FindRole.

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

Most-posted roles

View all roles at Walmart

More like this

Similar roles

Senior Data Scientist: Associate AI Experience

Walmart

(Usa) Bentonville Global Tech Ar Bentonville Home Office, US 35 days ago $90,000–$180,000
Python scikit-learn XGBoost LightGBM PyTorch TensorFlow Airflow Dagster Prefect FastAPI BentoML TorchServe Docker Kubernetes CI/CD Prometheus SQL LLM LangChain RAG FAISS

Principal Associate, Data Scientist - AI for Data

Capital One Financial

San Jose, Ca, US 69 days ago $161,800–$184,600
Python SQL TensorFlow PyTorch Natural-Language-Processing Information-Retrieval Search Recommendations LLM-Fine-Tuning

AI Data Scientist

Booz Allen Hamilton

Locations Fort Meade, Maryland, US 62 days ago $99,000–$225,000
Python Tableau Power BI Kubernetes AWS Azure GCP Docker CI/CD PostgreSQL MySQL MongoDB Git Jupyter Notebook Scikit-learn TensorFlow PyTorch REST APIs Swagger

Applied AI Data Scientist

The Hartford

Hartford, CT, US 9 days ago $90,160–$135,240
Python SQL PyTorch TensorFlow scikit-learn RAG pipelines GCP Vertex AI AWS SageMaker Bedrock BERTScore BLEURT Snowflake Vector Search Agentic AI LLM platforms Evaluation methodologies Semantic retrieval Feature engineering

Data Scientist AI/ML Lead

Lam Research

Fremont, Ca,Us, US 40 days ago $166,000–$350,000
Python TensorFlow PyTorch Databricks Azure ML Pandas PySpark Kubernetes CI/CD MLOps LLM RAG Transformer Ontology Knowledge_Graphs Semantic_Modeling Cloud_Infrastructure Data_Analysis Feature_Engineering

Principal Data & AI Engineer

Adobe

San Jose, US 73 days ago $208,300–$301,600
AWS Python Apache Airflow DynamoDB MySQL SageMaker OpenSearch Pinecone LangChain LangGraph MLFlow CI/CD Docker Kubernetes PySpark