As a Distinguished Data Scientist at Walmart's Colony Platform team, you will provide technical leadership for next-generation AI systems and platforms, driving breakthrough system design and advanced solution development that transforms the company at scale. Your daily responsibilities include architecting end-to-end ML/AI systems from feature engineering to production deployment, building autonomous AI agents, and deploying NLP pipelines using BERT/SBERT embeddings and vector search (FAISS). You will also mentor senior talent, establish best practices for model validation, experimentation, and safe deployment, and influence the long-term strategy for AI capabilities. The role requires deep expertise in ML algorithms, deep learning frameworks like PyTorch, proficiency with multi-cloud platforms such as AWS SageMaker and GCP Vertex AI, and strong Python skills including Pandas, NumPy, and Scikit-learn. You will work on high-impact projects that span NLP, computer vision, time series forecasting, and multi-agent AI systems at enterprise scale, ensuring measurable business value across Walmart’s global operations.
Skills
Python
Pandas
NumPy
Scikit-learn
Docker
MLOps
AWS SageMaker
GCP Vertex AI
BigQuery
BigQuery ML
XGBoost
CatBoost
LightGBM
Random Forest
AutoML
PyTorch
BERT
SBERT
FAISS
RAG
YOLO
CLIP
ARIMA
Prophet
LangChain
Pydantic AI
NLP
Computer Vision
Time Series Forecasting
What you'll do
Build and deploy production ML systems end-to-end, including data pipelines and monitoring.
Provide technical vision and lead advanced development in agentic AI and reinforcement learning.
Design and deploy large-scale AI systems and autonomous agents using LangChain and Pydantic AI.
Establish best practices for model validation, experimentation, and safe deployment of AI systems.
Mentor senior scientists and engineers on advanced modeling techniques and feature engineering.
What we're looking for
Deep expertise in ML algorithms and deep learning frameworks.
Hands-on experience building and deploying autonomous AI agents.
Proficiency with multi-cloud ML platforms (AWS SageMaker, GCP Vertex AI).
Strong Python skills including containerization and MLOps practices.
Advanced statistical analysis and causal inference capabilities.
Experience translating complex analytical results into business recommendations.
Market check
Salary context
This
$130,000–$260,000
range sits above
78%
of similar postings on FindRole.
Peer median band
$110,000–$220,000
Median floor and ceiling across peers.
Typical midpoint (25–75%)
$126,800–$188,050
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.
(Usa) Sunnyvale Tech Corners Ca Sunnyvale Home Office, US120 days ago
$169,000–$338,000
Python
SQL
LLMs
Generative AI
Agentic AI
Neural Search
Information Retrieval
Ranking Systems
ML Frameworks
Cloud-Native Infrastructure
Distributed Systems
CI/CD