Staff ML Engineer: Lead Scalable AI & LLM Systems

Walmart Inc.

Bentonville (AR)

On-site

USD 130,000 - 260,000

Full time

14 days+

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Benefits offered by this job

Health benefits
Live Better U education program
PTO and paid time off
Stock purchase plan
Employee discounts

Job summary

Walmart Inc. is seeking a Staff Machine Learning Engineer in Bentonville, AR to lead the design and deployment of scalable, production-grade ML and AI solutions across cloud environments.

You will own the full ML lifecycle from data engineering to monitoring, mentor teams, and drive improvements using advanced techniques like AutoML and NLP. You will work with Walmart Data Ventures and the Scintilla team to advance generative AI, agentic intelligence, and measurement-driven innovation that

Qualifications

  • Bachelor's degree in CS/CE/related field and 4+ years in software/ML engineering.

Responsibilities

  • Lead design and delivery of scalable, production-grade ML and AI systems.
  • Translate business problems into technical requirements, architectures, and success criteria.
  • Productionize DS prototypes across batch, real-time, and agentic workloads.
  • Own ML lifecycle, SDLC, CI/CD, versioning, testing, deployment, evaluation, observability, and monitoring.
  • Design systems balancing model quality, latency, throughput, reliability, scalability, maintainability, and cost.
  • Build reusable ML infrastructure, frameworks, and platform capabilities.
  • Lead production LLM/agentic systems including retrieval, toolcalling, orchestration, and state management.
  • Establish offline and online evaluation practices to detect regressions, drift, quality degradation, and production failure modes.
  • Drive architecture reviews, engineering standards, mentorship, and continuous innovation.

Skills

Python
SQL
System design
CI/CD
MLOps
Cloud-native architectures
APIs
Containers
LLMs
Mentorship

Education

Bachelor's degree in computer science/engineering or related field
4+ years experience in software/machine learning engineering
Master's degree (preferred)

Tools

Docker
Kubernetes

Job description

Walmart Inc. is seeking a Staff Machine Learning Engineer in Bentonville, AR to lead the design and deployment of scalable, production-grade ML and AI solutions across cloud environments.

You will own the full ML lifecycle from data engineering to monitoring, mentor teams, and drive improvements using advanced techniques like AutoML and NLP. You will work with Walmart Data Ventures and the Scintilla team to advance generative AI, agentic intelligence, and measurement-driven innovation that

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Staff ML Engineer — Scalable AI Systems Leader
Staff ML Engineer — Scalable AI Systems Leader

Walmart Data Ventures • Bentonville (AR)

On-site
USD 130,000 - 260,000
Health benefits
401(k)
Stock purchase
+2
Senior ML Engineer for AI Search & Scalable ML Systems
Senior ML Engineer for AI Search & Scalable ML Systems

Walmart • Sunnyvale (CA)

On-site
USD 143,000 - 286,000
Stock options
Education benefits
Paid time off
Lead ML Architect for Scalable Production AI Systems
Lead ML Architect for Scalable Production AI Systems

Walmart Inc. • Sunnyvale (CA)

On-site
USD 169,000 - 338,000
Health benefits
401(k)
Stock purchase
+2
Staff Software Engineer — AI & Data Platforms
Staff Software Engineer — AI & Data Platforms

Walmart Inc. • Bentonville (AR)

On-site
USD 110,000 - 220,000
Health benefits
401(k) and stock purchase
Principal Software Engineer: AI/ML & Cloud Lead
Principal Software Engineer: AI/ML & Cloud Lead

Walmart • Bentonville (AR)

On-site
USD 95,000 - 165,000
Health benefits
401(k)
Stock purchase
+3
Lead Software Engineer: AI-Enhanced Systems & Scalability
Lead Software Engineer: AI-Enhanced Systems & Scalability

Walmart, Inc. • Bentonville (AR)

On-site
USD 90,000 - 180,000
Performance-based bonus
Health benefits
Stock purchase plan
+3
Staff Software Engineer: AI & Cloud Platform Lead
Staff Software Engineer: AI & Cloud Platform Lead

StartupNWA • Bentonville (AR)

Hybrid
USD 110,000 - 220,000
Health benefits
401(k) and stock purchase plan
PTO including sick leave and holidays
+1
Principal AI Scientist – Lead Large-Scale ML
Principal AI Scientist – Lead Large-Scale ML

Walmart Inc. • Bentonville (AR)

On-site
USD 110,000 - 220,000
Performance bonus
Stock options
Comprehensive benefits
Staff Machine Learning Engineer - Scale AI Systems & Teams
Staff Machine Learning Engineer - Scale AI Systems & Teams

Walmart • Sunnyvale (CA)

On-site
USD 169,000 - 338,000
Stock purchase plan
Health benefits
Paid time off + education benefits
Principal AI/ML Platform Engineer
Principal AI/ML Platform Engineer

Walmart Inc. • Bentonville (AR)

On-site
USD 95,000 - 165,000
Health benefits
401(k) plan
Stock purchase program
+1