Principal Machine Learning Engineer

Walmart

. Architect and guide scalable, production-grade machine learning systems for customer-facing and operational products .

Posted 9/24/2026full-timeSunnyvale • California • United StatesLead💰 $169,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 scalable machine learning systems and managing the end-to-end ML lifecycle, including data sourcing, model training, and deployment. Proven ability to mentor teams, influence architectural decisions, and ensure compliance with responsible AI standards.

Highest-signal resume keywords

Machine Learning System ArchitectureEnd-to-End ML Lifecycle ManagementDeep Learning ExpertiseNLP ProficiencyMLOps 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 EngineeringDeep LearningNLPReinforcement LearningPythonJavaSQLTensorFlowPyTorchScikit-learn

Soft Skills

MentoringCross-Functional CollaborationCritical ThinkingTechnical Writing

Industry Keywords

Model EvaluationFairnessResponsible AIFeature EngineeringHyperparameter OptimizationTime Series Analysis

Tech Stack

Tools & technologies

JavaPythonPyTorchScikit-LearnSQLTensorflow

About the role

Key responsibilities & impact

Architect and guide scalable, production-grade machine learning systems for customer-facing and operational products
Lead end-to-end ML lifecycle execution, including data sourcing, feature engineering, model training, deployment, and monitoring
Drive innovation through deep learning, NLP, reinforcement learning, and multimodal modeling approaches
Mentor engineering teams and influence architectural decisions
Collaborate cross-functionally to align ML solutions with platform strategies and downstream applications
Establish standards for model evaluation, fairness, and responsible AI compliance
Publish technical insights and contribute to Walmart’s engineering knowledge base

Requirements

What you’ll need

Bachelor's degree in computer science, computer engineering, computer information systems, software engineering, or related area and 5 years’ experience in software engineering, machine learning engineering, AI systems or related area; OR 7 years’ experience in those areas
2 years’ supervisory experience
Extensive experience designing and architecting scalable, production-grade machine learning systems
Proficiency in TensorFlow, PyTorch, Scikit-learn, Python, Java, and SQL
Expertise in end-to-end ML lifecycle management
Strong background in deep learning, NLP, reinforcement learning, and time series analysis
Ability to lead technical strategy, mentor engineers, and influence cross-functional collaboration
Skill in ML model validation, hyperparameter optimization, and MLOps best practices
Excellent critical thinking and technical writing capabilities

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
Live Better U Walmart-paid education benefit
Tuition, books, and fees completely paid for eligible education programs