Staff Machine Learning Engineer

Walmart

. Lead design and delivery of scalable, production-grade ML and AI systems .

Posted 9/16/2026full-timeBentonville • California • United StatesLead💰 $130,000 - $338,000 per yearWebsite

Core Competencies

Role fit

Core Competencies

Use this summary to align your resume positioning with the role.

Demonstrates expertise in leading the design and delivery of scalable ML and AI systems, with a strong focus on MLOps, system design, and production operations. Proficient in Python and SQL, with a solid foundation in algorithms and machine learning methods.

Highest-signal resume keywords

Machine Learning EngineeringMLOpsPython ProgrammingSystem DesignTechnical Leadership

ATS Keywords

Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills

Machine LearningAI SystemsSoftware EngineeringData StructuresAlgorithmsCI/CDProduction OperationsBatch ProcessingReal-Time InferenceObservability

Soft Skills

MentorshipCross-Functional Communication

Tools & Technologies

Cloud-Native ArchitecturesAPIsContainersRelational DatabasesNoSQL Databases

Industry Keywords

ML LifecycleScalabilityModel QualityLatencyThroughput

Tech Stack

Tools & technologies

CloudDistributed SystemsNoSQLPythonSDLCSQL

About the role

Key responsibilities & impact

Lead design and delivery of scalable, production-grade ML and AI systems
Translate business problems into technical requirements, architectures, and measurable success criteria
Productionize data science prototypes across batch, real-time, and agentic workloads
Own the 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 and agentic systems, including retrieval, tool calling, 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
Provide technical leadership and mentor engineering teams
Support Walmart Data Ventures’ data and AI platform for suppliers and retail outcomes

Requirements

What you’ll need

Bachelor's degree in computer science, computer engineering, computer information systems, software engineering, or related area and 4 years’ experience in software engineering, machine learning engineering, AI systems or related area; OR 6 years’ experience in those areas
Proficiency in Python and SQL
Strong software engineering fundamentals
System design, testing, CI/CD, and production operations
Extensive experience designing and operating large-scale ML systems across training, evaluation, batch and real-time inference, monitoring, and continuous improvement
Strong understanding of MLOps, distributed systems, cloud-native architectures, APIs, containers, and relational/NoSQL data platforms
Experience building reusable ML infrastructure, frameworks, and observability capabilities
Strong foundation in algorithms, data structures, and machine learning methods
Experience with LLM and agentic systems, including retrieval, tool calling, orchestration, and evaluation
Ability to balance model quality, latency, throughput, reliability, scalability, maintainability, and cost
Demonstrated technical leadership, mentorship, and effective cross-functional communication

Benefits

Comp & perks

Performance-based bonus awards
Medical, vision and dental coverage
401(k)
Stock purchase
Company-paid life insurance
PTO, including sick leave
Parental leave
Family care leave
Bereavement leave
Jury duty leave
Voting leave
Short-term and long-term disability
Company discounts
Military Leave Pay
Adoption and surrogacy expense reimbursement
PTO and/or PPTO for vacation, sick leave, holidays, or other purposes
Live Better U Walmart-paid education benefit program
High school completion, bachelor's degrees, English Language Learning, and short-form certificate programs
Tuition, books, and fees completely paid for by Walmart
Stock
Annual or quarterly performance bonuses