Principal, Data Scientist -- Agentic AI

Walmart

Quick summary

Work type
On-site
Location
Sunnyvale, CA
Salary
$143,000–$286,000 / yr
Posted
71 days ago

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $206k
This role $214k
$126k most similar roles pay here $303k

This role pays more than 54% of similar roles. Most pay $165,375–$246,150 — the shaded band above. At the midpoint, this role pays about $214k versus about $206k 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 529 open roles on FindRole.

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

Most-posted roles

View all roles at Walmart

At a glance

TL;DR · Principal, Data Scientist -- Agentic AI

Join Walmart Data Ventures as a Principal Data Scientist on their pioneering team driving the future of retail AI. You will architect and scale next-generation ML and LLM systems, integrating advanced capabilities like planning, memory, tool-use, and multi-agent collaboration into Walmart’s digital ecosystem. Your day-to-day involves defining best practices for LLMOps, conducting applied research in large language models and multimodal AI, and building end-to-end ML systems with a focus on fairness, explainability, and responsible AI compliance. Essential skills include deep expertise in Generative AI, LLMs, Agentic AI, strong programming in Python with PyTorch/TensorFlow, and experience in MLOps practices. This role demands leadership in technical mentoring and cross-functional strategy influence, contributing to Walmart’s reputation through thought leadership. Ideal candidates are curious, resilient, and proficient in cutting-edge ML technologies, ready to lead the next era of AI innovation at scale for global impact.

What you'll do

  • Architect agentic AI systems with capabilities like planning, memory, tool-use, and multi-agent collaboration.
  • Define best practices for LLMOps including prompt versioning, embedding store management, and synthetic data generation.
  • Drive applied research in large language models (LLMs) and multimodal AI using advanced methods like LoRA/PEFT fine-tuning.
  • Build end-to-end ML systems covering data sourcing, feature engineering, model training, deployment, monitoring, and iteration.
  • Champion MLOps practices such as automated pipelines, CI/CD for ML, model versioning, drift detection, and telemetry-based monitoring.

What we're looking for

  • Deep expertise in Generative AI, LLMs, and Agentic AI systems.
  • Proven experience architecting large-scale ML/AI platforms with scalability focus.
  • Strong programming skills in Python, PyTorch/TensorFlow for ML development.
  • Track record of technical leadership and mentoring senior engineers/scientists.
  • Familiarity with agentic development platforms like Google ADK, LangGraph, CrewAI.

More like this

Similar roles

Principal Data Scientist: Associate AI experience

Walmart

Bentonville, AR 44 days ago $110,000$220,000
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

Principal Applied Scientist, Agentic AI

Zillow

Remote (Remote-Usa, US) 75 days ago $191,300$305,700
ReinforcementLearning PostTraining RLHF Rlaif DPO SupervisedFineTuning PreferenceModeling MultiObjectiveOptimization LLMs MultimodalModels VectorSearch OrchestrationFrameworks Python TensorFlow PyTorch CI/CD AWS Kubernetes GitHub JupyterNotebooks
Remote

Senior Data Scientist - Agentic AI

Allstate

Remote (Illinois) 64 days ago $100,000$170,500
Python TensorFlow PyTorch R SQL AWS Azure CI/CD Docker Prometheus Grafana Infrastructure as Code AWS Bedrock Azure Foundry AI Agents LangGraph OpenAI Agents SDK Cloud Security Machine Learning Data Science
Remote

Principal Machine Learning Engineer, Agentic AI

Zillow

Remote (Remote-Usa, US) 114 days ago $204,400$326,600
Python AgentSDK LangChain LangGraph GenAI ReinforcementLearning A/BTesting CI/CD LLM OpenAI PostgreSQL Kubernetes Docker Prometheus Grafana
Remote

Principal Machine Learning Engineer, Agentic AI

Zillow

Remote (Remote-Usa, US) 73 days ago $204,400$326,600
Python TensorFlow PyTorch Kubernetes AWS Docker CI/CD PostgreSQL LangGraph Agents SDK AutoGen Prometheus Grafana Scalability Fault_Tolerance
Remote