Staff Machine Learning Engineer

Walmart

. Architect and lead development of scalable, production-grade ML systems .

Posted 10/13/2026full-timeSunnyvale • 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 architecting and leading the development of scalable machine learning systems, with a strong focus on the end-to-end ML lifecycle, MLOps practices, and integration into financial processes. Proficient in deep learning, natural language processing, and model optimization, while also mentoring teams and influencing architectural decisions.

Highest-signal resume keywords

End-To-End ML Lifecycle ManagementScalable Machine Learning Systems ArchitectureTensorFlowDeep LearningMLOps Best Practices

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 Learning EngineeringAI SystemsPythonSQLJavaNeural NetworksHyperparameter OptimizationModel ValidationFeature EngineeringModel Training

Soft Skills

Critical ThinkingTechnical WritingMentoring

Tools & Technologies

TensorFlowPyTorchScikit-learn

Industry Keywords

MLOpsNLPReinforcement LearningFinancial ReconciliationDigital Accessibility

Tech Stack

Tools & technologies

JavaPythonPyTorchScikit-LearnSQLTensorflow

About the role

Key responsibilities & impact

Architect and lead development of scalable, production-grade ML systems
Execute the end-to-end ML lifecycle, including data sourcing, feature engineering, model training, deployment, and monitoring
Integrate machine learning systems into customer/associate-facing and operational products
Translate complex business challenges into technical strategies, platform requirements, and success metrics
Champion MLOps practices, including CI/CD pipelines, model versioning, drift detection, and performance telemetry
Drive innovation in deep learning, NLP, reinforcement learning, and multimodal modeling
Mentor engineers and data science teams
Influence architectural decisions and promote engineering excellence
Collaborate cross-functionally to align ML solutions with platform strategies and ensure scalability
Develop AI and ML solutions for financial processes, including anomalous sales posting, cash loss management, and financial reconciliation

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 software engineering, machine learning engineering, AI systems or related area
Strong expertise in end-to-end ML lifecycle management
Extensive experience architecting scalable, production-grade machine learning systems and platforms
Proficiency with TensorFlow, PyTorch, Scikit-learn, Python, SQL, and Java
Deep knowledge of neural networks, deep learning, natural language processing, reinforcement learning, and time series analysis
Hands-on experience building systems supporting agentic architectures, including memory, trace, eval, orchestration, and routing layers
Skill in hyperparameter optimization, model validation, and MLOps best practices
Excellent critical thinking and technical writing capabilities
Preferred: Experience with Computer Vision technologies and/or Finance domain
Preferred: Master’s degree in a related area and 2 years' experience in software engineering, machine learning engineering, AI systems or related area
Preferred: Experience with accessibility best practices, inclusive digital experiences, WCAG 2.2 AA standards, assistive technologies, and digital accessibility

Benefits

Comp & perks

Competitive pay
Performance-based bonus awards
Medical, vision, and dental coverage
401(k)
Stock purchase
Company-paid life insurance
Paid time off, including sick leave
Parental leave
Family care leave
Bereavement leave
Jury duty leave
Voting leave
Short-term disability
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 education benefit: Walmart-paid high school completion, bachelor's degree, English Language Learning, and short-form certificate programs
Tuition, books, and fees completely paid by Walmart
Stock may be included as additional compensation for certain positions