Senior AI Systems Architect & Production Innovator

Walmart; Bentonville (AR); On-site; USD 110,000 - 220,000; Full time

Benefits offered by this job

Health, vision, dental benefits

401(k) and stock purchase program

PTO and parental leave

Job summary

Walmart seeks a Principal Data Scientist to lead and shape next‑gen AI platforms at scale. You will architect end‑to‑end ML/AI systems, build autonomous AI agents, and mentor senior engineers across the Associate AI Experiences team.

You will blend deep statistical rigor with practical business impact, deploying enterprise‑grade NLP, CV, and forecasting solutions while guiding technical standards and ROI-focused outcomes.

Qualifications

• Master's degree in a relevant field with 5+ years of impact-focused data science/ML experience.
• Experience building/producing production ML systems, NLP, CV, or time-series projects.
• PhD preferred with related industry experience.

Responsibilities

• Build and deploy production ML systems end-to-end, including data pipelines and monitoring.
• Lead advanced development in agentic AI, RL, and simulation.
• Architect and deploy large-scale AI systems and autonomous agents with tool-calling and multi-agent orchestration.
• Present insights to leadership with clear data visualizations and business impact.

Skills

ML algorithms

NLP

Agentic AI

Python

MLOps

Experiment design

Causal inference

Production systems

Mentoring

System architecture

Education

Master's degree in Statistics, Economics, Analytics, Mathematics, CS, IT

PhD in AI/ML or related field

Tools

LangChain

Pydantic AI

RAG

AWS SageMaker

GCP Vertex AI

BigQuery

FAISS

Docker

Job description

Walmart seeks a Principal Data Scientist to lead and shape next‑gen AI platforms at scale. You will architect end‑to‑end ML/AI systems, build autonomous AI agents, and mentor senior engineers across the Associate AI Experiences team.

You will blend deep statistical rigor with practical business impact, deploying enterprise‑grade NLP, CV, and forecasting solutions while guiding technical standards and ROI-focused outcomes.