Lead the design and development of scalable, production-grade machine learning systems to solve complex business problems.

Architect and guide development of scalable, production-grade ML systems integrated into customer-facing and operational products
Own end-to-end ML lifecycle execution: data sourcing, feature engineering, model training, deployment, and monitoring
Drive applied ML innovation using advanced modeling techniques including deep learning, NLP, reinforcement learning, and multimodal approaches
Mentor engineering teams, influence architectural decisions, and promote technical excellence across ML disciplines
Collaborate cross-functionally to align ML solutions with platform strategies and ensure integration with downstream applications
Establish standards for model evaluation, fairness, and responsible AI compliance
Publish technical insights and contribute to Walmart’s engineering knowledge base

Extensive experience designing and architecting scalable, production-grade ML systems aligned with business objectives
Proficiency with ML frameworks: TensorFlow, PyTorch, Scikit-learn; programming languages including Python, Java, and SQL
Expertise in end-to-end ML lifecycle management: data sourcing, feature engineering, model training, deployment, and monitoring
Strong background in advanced modeling techniques: deep learning, NLP, reinforcement learning, and time series analysis
Proven ability to lead technical strategy, mentor engineers, and partner effectively across teams
Skilled in ML model validation, hyperparameter optimization, and MLOps best practices
Strong critical thinking and technical writing capabilities to communicate complex concepts clearly

Option 1: 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
Option 2: 7 years’ experience in software engineering, machine learning engineering, AI systems, or related area
2 years’ supervisory experience

Master’s degree in computer science, computer engineering, computer information systems, software engineering, or related area and 3 years’ experience in software engineering, machine learning engineering, AI systems, or related area
Experience applying accessibility best practices
Experience creating inclusive digital experiences, including knowledge of implementing WCAG 2.2 AA standards, assistive technologies, and integrating digital accessibility

International Recommendations Platform team builds a multi-tenant framework delivering personalized experiences across Walmart International sites
Pioneers innovation in Machine Learning and AI to improve customer engagement and business outcomes
Uses cloud-native architecture and software design patterns to deliver scalable, resilient solutions
Emphasizes collaboration across API design, system reviews, and data-driven decisions
Focuses on continuous improvement via DevOps engineering and software prototyping for accelerated delivery and high-quality standards

815 Eleventh Ave, Sunnyvale, CA 94089-4731, United States of America (onsite)

USD 169,000 - 338,000 per yearly

Competitive pay
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
Jury duty
Voting
Short-term and long-term disability
Company discounts
Military Leave Pay
Adoption and surrogacy expense reimbursement
Live Better U (Walmart-paid education benefit program for full-time and part-time associates)

Principal, Machine Learning Engineer

Job Description