(USA) Principal, Machine Learning Engineer

Walmart Inc.

Sunnyvale (CA)

On-site

USD 169,000 - 338,000

Full time

22 hours ago
Be an early applicant
Application generator

Don’t send a generic resume — generate a resume and cover letter tailored to this exact role.

Get past ATS filters

Benefits offered by this job

Health benefits
401(k)
Stock purchase
Paid time off
Live Better U

Job summary

Walmart seeks a Principal, Machine Learning Engineer in Sunnyvale, CA to design scalable, production-grade ML systems and lead end-to-end ML lifecycle execution. You will mentor engineers and collaborate across teams to deliver ML-powered customer-facing and operational solutions.

The role requires expertise in NLP, deep learning, time series, and responsible AI, with a strong emphasis on deployment, monitoring, and cross-functional impact.

Qualifications

  • Minimum qualifications include a Bachelor's degree in CS/CE/related field with 5 years' software/machine learning experience.
  • Option 2: 7 years' experience in software engineering, machine learning engineering, AI systems or related area and 2 years supervisory experience.
  • Preferred qualifications include Master's degree and WCAG accessibility experience.

Responsibilities

  • Architect and guide the development of scalable, production-grade machine learning systems integrated into customer-facing and operational products.
  • Lead end-to-end ML lifecycle execution, including data sourcing, feature engineering, model training, deployment, and monitoring.
  • Drive innovation in applied ML through advanced modeling techniques such as 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 seamless integration with downstream applications.
  • Establish standards for model evaluation, fairness, and responsible AI compliance.

Skills

TensorFlow
PyTorch
Scikit-learn
Python
Java
SQL
ML lifecycle
MLOps
Leadership

Education

Bachelor's degree in Computer Science / Engineering / related field
5+ years of software/machine learning engineering experience

Tools

TensorFlow
PyTorch
Scikit-learn

Job description

(USA) Principal, Machine Learning Engineer

Sunnyvale, CA

(USA) Principal, Machine Learning Engineer

811 11 Th Ave
Sunnyvale, CA 94089-4731

R-2640035

$169,000 - $338,000/yr

Regular/ Permanent

Position Summary...
What you'll do...
Role summary:
The Principal, Machine Learning Engineer will lead the design and development of scalable, production-grade ML systems that address complex business challenges. This role involves architecting end-to-end ML pipelines, driving innovation in advanced modeling techniques, and ensuring robust deployment and monitoring practices. The position requires collaboration with cross-functional teams to align ML solutions with organizational goals while mentoring technical staff and promoting engineering excellence. The successful candidate will apply sound technical judgment, foster continuous improvement, and contribute to Walmart's reputation through technical leadership and knowledge sharing.
About the team:
The International Recommendations Platform team develops a multi-tenant framework delivering personalized experiences across Walmart International sites. The team pioneers innovation in Machine Learning and AI to improve customer engagement and business outcomes. Utilizing cloud-native architecture and software design patterns, members build scalable, resilient solutions. Collaboration is key, focusing on API design, system reviews, and data-driven decisions. The team emphasizes continuous improvement through DevOps engineering and software prototyping, ensuring accelerated delivery and high-quality standards in a dynamic environment. This environment fosters technical excellence and impactful innovation.
What you'll do:
  • Architect and guide the development of scalable, production-grade machine learning systems integrated into customer-facing and operational products.
  • Lead end-to-end ML lifecycle execution, including data sourcing, feature engineering, model training, deployment, and monitoring.
  • Drive innovation in applied ML through advanced modeling techniques such as 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 seamless 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.

What you'll bring:
  • Extensive experience designing and architecting scalable, production-grade machine learning systems aligned with business objectives.
  • Proficiency in ML frameworks such as TensorFlow, PyTorch, Scikit-learn, and programming languages including Python, Java, and SQL.
  • Expertise in end-to-end ML lifecycle management, including data sourcing, feature engineering, model training, deployment, and monitoring.
  • Strong background in advanced modeling techniques, including deep learning, NLP, reinforcement learning, and time series analysis.
  • Demonstrated ability to lead technical strategy, mentor engineers, and influence cross-functional collaboration.
  • Skilled in ML model validation, hyperparameter optimization, and MLOps best practices.
  • Excellent critical thinking and technical writing capabilities to communicate complex concepts clearly.
At Walmart, we offer competitive pay as well as performance-based bonus awards and other great benefits for a happier mind, body, and wallet. Health benefits include medical, vision and dental coverage. Financial benefits include 401(k), stock purchase and company-paid life insurance. Paid time off benefits include PTO (including sick leave), parental leave, family care leave, bereavement, jury duty, and voting. Other benefits include short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement, and more.You will also receive PTO and/or PPTO that can be used for vacation, sick leave, holidays, or other purposes. The amount you receive depends on your job classification and length of employment. It will meet or exceed the requirements of paid sick leave laws, where applicable. For information about PTO, see https://one.walmart.com/notices .Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities. Programs range from high school completion to bachelor's degrees, including English Language Learning and short-form certificates. Tuition, books, and fees are completely paid for by Walmart.
Eligibility requirements apply to some benefits and may depend on your job classification and length of employment. Benefits are subject to change and may be subject to a specific plan or program terms.
For information about benefits and eligibility, see One.Walmart . The annual salary range for this position is $169,000.00 - $338,000.00Additional compensation includes annual or quarterly performance bonuses.Additional compensation for certain positions may also include :
- Stock
Minimum Qualifications...

Outlined below are the required minimum qualifications for this position. If none are listed, there are no minimum qualifications.

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.

Preferred Qualifications...

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., Prior work experience in accessibility best practices., Prior work experience in creating inclusive digital experiences, demonstrating knowledge in implementing Web Content Accessibility Guidelines (WCAG) 2.2 AA standards, assistive technologies, and integrating digital accessibility seamlessly.

Primary Location...

815 Eleventh Ave, Sunnyvale, CA 94089-4731, United States of AmericaWalmart and its subsidiaries are committed to maintaining a drug-free workplace and has a no tolerance policy regarding the use of illegal drugs and alcohol on the job. This policy applies to all employees and aims to create a safe and productive work environment.

Walmart is committed to maintaining a drug-free workplace and has a no tolerance policy regarding the use of illegal drugs and alcohol on the job. This policy applies to all employees and aims to create a safe and productive work environment.

Benefits you'll enjoy

Discount Card

Get 10% off

Walmart associates are eligible for a 10% discount card on most regular-priced items and fresh produce in-store and on select items at Walmart.com. Eligible after 90 days of employment.

Live Better U

100% covered

Earn a degree or in-demand skills certificates with no debt- Walmart covers 100% of tuition and books. Live Better U offers 60+ programs for Associates to pursue their dreams.

Ready to grow your career? Walmart Academy offers job-specific retail training and leadership courses to help Associates reach their career goals.

Medical, dental, vision, and wellness programs for you and your family

Access mental health resources and assistance programs for life's challenges

Training, leadership programs, and clear paths to advance.

Imagine a workplace surrounded by innovation.At Walmart's new Home Office in Bentonville, Arkansas, we're redefining what it means to work at a global leader.

Set on 350 acres of thoughtfully revitalized land, our new campus seamlessly integrates the charm of Northwest Arkansas with cutting edge design and technology. From biking trails and outdoor courtyards to flexible, tech-enabled workspaces, every detail reflects our commitment to sustainability, connection, and culture.

With amenities like on-site childcare at our Little Squiggles Children's Enrichment Center, the Walton Family Whole Health & Fitness Center, and a vibrant food hall featuring local and international favorites, we're creating a space where work-life balance isn't just a goal—it's a reality.
Beyond the campus, Bentonville offers a dynamic lifestyle with world-class dining, art at the Crystal Bridges Museum, and countless outdoor activities.

Whether you're exploring High South cuisine, enjoying live performances at our new amphitheater, or cycling the Razorback Greenway, you'll experience the perfect blend of small-town charm and big-city amenities.

Join us at Walmart and grow your career alongside a community that feels like home.

This isn't just a workplace—it's a destination for leaders eager to make a difference.

Trending roles

(USA) Principal, Machine Learning Engineer

Sunnyvale, CA + 2 more

(USA) Principal, Software Engineer

Sunnyvale, CA 94089-1114

Principal, Software Engineer

Sunnyvale, CA + 1 more

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Principal, Software Engineer
Principal, Software Engineer

Walmart Inc. • Sunnyvale (CA)

On-site
USD 143,000 - 286,000
Discount Card
Live Better U education benefits
401(k) and stock purchase
+2
(USA) Principal, Software Engineer
(USA) Principal, Software Engineer

Walmart Inc. • Sunnyvale (CA)

On-site
USD 143,000 - 286,000
Discount Card
Live Better U 100% covered
Walmart Academy training
Distinguished, Software Engineer
Distinguished, Software Engineer

Walmart Inc. • Sunnyvale (CA)

On-site
USD 169,000 - 338,000
Discount Card
Live Better U 100% covered
Medical, dental, vision and wellness
(USA) Principal, Data Scientist
(USA) Principal, Data Scientist

Walmart Inc. • Sunnyvale (CA)

On-site
USD 143,000 - 286,000
Health benefits
401(k)
Stock purchase plan
+1
(USA) Principal, Software Engineer
(USA) Principal, Software Engineer

Walmart Inc. • Hoboken (NJ)

On-site
USD 132,000 - 264,000
Medical, dental, vision coverage
401(k) and stock purchase
PTO including sick leave and holidays
Principal, Technical Program Manager
Principal, Technical Program Manager

Walmart Inc. • Sunnyvale (CA)

On-site
USD 143,000 - 286,000
Health benefits
401(k) match
Stock purchase plan
+2
(USA) Staff, Machine Learning Engineer
(USA) Staff, Machine Learning Engineer

Walmart Inc. • Bentonville (AR)

On-site
USD 130,000 - 260,000
Health benefits
Live Better U education program
PTO and paid time off
+2
(USA) Principal, Software Engineer
(USA) Principal, Software Engineer

Walmart Inc. • Bellevue (WA)

On-site
USD 132,000 - 264,000
Health benefits
401(k) and stock purchase
Principal Software Engineer - Tech
Principal Software Engineer - Tech

Walmart Inc. • Bentonville (AR)

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

Walmart Inc. • Bellevue (WA)

On-site
USD 156,000 - 312,000
Discount Card
Live Better U education benefit
Stock purchase program
+1