About the role

Data Scientist building production ML systems and deploying autonomous AI agents for Walmart. Collaborating across teams to drive business insights from complex data and enhance operational efficiency.

Responsibilities

Build and deploy production ML systems end‑to‑end, including data pipelines, feature stores, model training, serving layers, monitoring, and feedback loops at scale.
Architect and deploy autonomous AI agents, including conversational assistants, predictive agents with tool calling, and multi‑agent orchestration systems using LangChain, Pydantic AI, and RAG patterns.
Design agentic evaluation frameworks to benchmark agent performance across task completion, code quality, and multi‑step reasoning accuracy.
Deploy NLP pipelines at enterprise scale, enabling semantic similarity across millions of records using BERT/SBERT embeddings and vector search (FAISS).
Engineer computer vision systems for real‑time inference (YOLO, RT‑DETR, CLIP) with multi‑GPU training optimization.
Build time‑series forecasting models that drive inventory optimization, demand planning, and financial projections across multiple geographies.
Present insights to senior leadership using compelling data visualizations, quantified business impact, and clearly defined confidence intervals.
Mentor team members on advanced modeling techniques, feature engineering, and experimental design.
Travel internationally to support Walmart operations in Canada, Chile, Mexico, South Africa, Europe, and the UK.

Option 1: Bachelors degree in Statistics, Economics, Analytics, Mathematics, Computer Science, Information Technology or related field and 5 years' experience in an analytics related field.
Option 2: Masters degree in Statistics, Economics, Analytics, Mathematics, Computer Science, Information Technology or related field and 3 years' experience in an analytics related field.
Option 3: 7 years' experience in an analytics or related field.
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 across AWS SageMaker and GCP (Vertex AI, BigQuery, BigQuery ML).
Strong Python skills (Pandas, NumPy, scikit‑learn), containerization expertise (Docker), and MLOps best practices.
Advanced knowledge of statistical analysis, experimental design, and causal inference.
Proven ability to translate complex analytical results into clear, actionable business recommendations for non‑technical stakeholders.

Health benefits include medical, vision and dental coverage.
Financial benefits include 401(k), stock purchase and company-paid life insurance.
Paid time off benefits include PTO (including sick leave), parental leave, family care leave, bereavement, jury duty, and voting.
Other benefits include short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement, and more.
Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities.