Hybrid Senior Data Scientist – Associate AI Experience

at Walmart

• Job title: Senior Data Scientist – Associate AI Experience
• Job type: Full Time
• Experience level: Senior
• Salary: $90,000 - $180,000 per year
• Degree requirement: Postgraduate Degree
• Tech skills: Airflow, Docker, Kubernetes, Python, PyTorch, Scikit-Learn, SQL, Tensorflow
• Location requirements: Hybrid, Bentonville, United States

About the role

• Senior Data Scientist providing technical leadership and vision for next-generation AI systems at Walmart. Driving advanced solution development to transform operations globally.

Responsibilities

• Architect end-to-end ML/AI systems — from feature engineering through production deployment and monitoring.
• Build and deploy production ML systems end-to-end: data pipelines, feature stores, model training, serving layers, monitoring, and feedback loops at scale.
• Provide technical visionand lead advanced development in agentic AI, reinforcement learning, and simulation.
• Architect and deploy large-scale AI systems and autonomous AI agents — conversational assistants, predictive agents with tool-calling, and multi-agent orchestration systems using LangChain, Pydantic AI, and RAG patterns.
• Design agentic evaluation frameworks that benchmark agent performance across task completion, code quality, and multi-step reasoning accuracy.
• Deploy NLP pipelines at enterprise scale — semantic similarity across millions of records using BERT/SBERT embeddings and vector search (FAISS).
• Lead high-impact projects end-to-end; problem framing, modeling, prototyping, production architecture, and rollout.
• Author and review technical designs and standards; present findings to senior leadership.
• Mentor senior scientists and engineers on advanced modeling techniques, feature engineering, and experimentation design - raising the technical bar across teams.
• Establish best practices for model validations, experiemention, and safe deployment of AI systems.
• Influence roadmap prioritization and long-term strategy for AI and simulation capabilities.
• Present insights to senior leadership with compelling data visualization, quantified business impact, and clear confidence intervals.

Requirements

• Experience building and deploying AI/ML solutions in production environments.
• Strong profiency in Python and common ML libraries (scikit-learn, XGBoost/LightGBM, PyTorch/TensorFlow).
• Demonstrated AI/ML engineering experience: Pipeline orchestration (Airflow, Dagster, Prefect) Model packaging/serving (FastAPI, BentoML, TorchServe) Containerization and deployment (Docker, Kubernetes) CI/CD and testing practices for AI/ML Monitoring, observability (model metrics, drift, loggin)
• Solid understanding of statistics, experimental design, and causal/measurement thinking.
• Experience with agentic AI / LLM systems including: Tool/function calling and agent orchestration framework (LangGraph/LangChain, Semantic Kernel, LlamaIndex, custom orchestrators).
• Prompt strategies, structured outputs, and reliability techniques RAG design and optimization.
• LLM/agent evaluation methodologies Strong SQL skills and experience working with large-scale datasets.
• Ability to translate complex analytical results into clear business recommendations for non-technical stakeholders.
• Masters degree in Statistics, Economics, Analytics, Mathematics, Computer Science, Information Technology, or related field AND 3+ years’ experience in data science or Machine Learning roles with demonstrated impact.

Benefits

• 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.
• You will also receive PTO and/or PPTO that can be used for vacation, sick leave, holidays, or other purposes.
• Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities. Programs range from high school completion to bachelor's degrees, including English Language Learning and short-form certificates. Tuition, books, and fees are completely paid for by Walmart. Eligibility requirements apply to some benefits and may depend on your job classification and length of employment.