(USA) Staff, Machine Learning Engineer

Walmart; Sunnyvale (CA); On-site; USD 169,000 - 338,000; Full time

Benefits offered by this job

Health benefits

401(k) plan

Stock purchase

Paid time off

Live Better U

Job summary

Walmart is seeking a Staff Machine Learning Engineer in Sunnyvale to lead the design and delivery of scalable, production-grade ML solutions. You will oversee data sourcing, model development, deployment, and monitoring while championing MLOps standards.

You'll mentor engineers and collaborate across Product, Data, and Platform teams to push innovation in NLP, CV, and deep learning. The role emphasizes reliability, scalability, and impact on business outcomes.

Qualifications

• Bachelor's degree in CS/engineering or related field and 4 years’ experience in software/machine learning.
• 6 years’ experience in software/ML or related area.

Responsibilities

• Lead the design and deployment of scalable, production-grade ML solutions.
• Translate complex problems into technical requirements and designs.
• Oversee the full ML lifecycle: data sourcing, feature engineering, training, deployment, monitoring.
• Implement MLOps best practices including CI/CD and model versioning.
• Build reusable ML infrastructure components and manage large-scale cloud operations.
• Provide technical leadership and mentorship to engineers.
• Drive experimentation pipelines and explore advanced ML techniques.

Skills

TensorFlow

PyTorch

Scikit-learn

Python

SQL

MLOps

Cloud platforms

Kubernetes

CI/CD

Education

Bachelor's degree

4+ years software/machine learning experience

6 years software/ML experience

Tools

Docker

Kubernetes

CI/CD pipelines

Telemetry/Monitoring

Job description

1345 Crossman Ave
Sunnyvale, CA 94089-1114

R-2594022

Loading map...

$169,000 - $338,000/yr

Regular/ Permanent

Position Summary...

The Staff Machine Learning Engineer will lead the design and delivery of scalable, production-grade ML solutions that address critical business needs. This role involves overseeing the end-to-end ML lifecycle, including data sourcing, model development, deployment, and monitoring, while applying MLOps best practices. The engineer will build and maintain ML infrastructure, drive technical standards, and mentor team members. Collaboration across teams and continuous innovation are essential to advance Walmart’s ML capabilities and ensure high-quality, reliable solutions that support business objectives and operational excellence.

About the team:

The team develops foundational AI models and services that enhance Walmart’s customer-focused applications. Collaborating with Product, Business, Data, and Platform groups, they apply expertise in predictive analytics, machine learning, and data visualization. Key responsibilities include identifying data sources, building scalable models, and ensuring data quality to support informed decisions. Leveraging advanced techniques in natural language processing, computer vision, and deep learning, the team strengthens Walmart’s data strategy and analytics capabilities. They foster continuous learning and innovation to deliver impactful, scalable solutions that drive business value and operational excellence.

What you'll do...

• Lead the design and deployment of scalable, production-grade machine learning solutions addressing critical business needs.
• Translate complex problems into clear technical requirements, architectural designs, and measurable success criteria.
• Oversee the full ML lifecycle, including data sourcing, feature engineering, model training, deployment, monitoring, and continuous improvement.
• Implement MLOps best practices such as CI/CD pipelines, model versioning, and monitoring to ensure operational excellence.
• Build and maintain reusable ML infrastructure components and manage large-scale operations across cloud environments using containerization and workflow automation.
• Provide technical leadership and mentorship, setting standards for code quality, testing, and deployment.
• Drive experimentation pipelines and explore advanced ML techniques to accelerate innovation and solution delivery.

What you'll bring:

• Proven expertise in designing and delivering scalable, production-grade machine learning solutions aligned with business needs.
• Strong proficiency in ML frameworks such as TensorFlow, PyTorch, and Scikit-learn, and coding languages including Python and SQL.
• Experience with end-to-end ML lifecycle management, including data sourcing, feature engineering, model training, deployment, and monitoring.
• Knowledge of MLOps best practices, including CI/CD pipelines, model versioning, and telemetry-based monitoring.
• Familiarity with cloud platforms (AWS, GCP, Azure), containerization, Kubernetes, and workflow automation.
• Ability to implement robust model evaluation and validation frameworks.
• Demonstrated leadership in mentoring engineers and enforcing engineering standards.
• Strong collaboration and communication skills to work across teams and stakeholders.

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.00 Additional 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 4 years’ experience in software engineering, machine learning engineering, AI systems or related area.
• Option 2: 6 years’ experience in software engineering, machine learning engineering, AI systems or related area.

Preferred Qualifications...

Outlined below are the optional preferred qualifications for this position. If none are listed, there are no preferred qualifications.

Master’s degree in computer science, computer engineering, computer information systems, software engineering, or related area and 2 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...

1345 Crossman Ave, Sunnyvale, CA 94089-1114, 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.

Enjoy 401(k) matching and stock purchase plans.

Paid time off

Take a break as needed for vacation, sick leave, holidays, parental leave, and more

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.