(USA) Senior, Machine Learning Engineer

Walmart; Sunnyvale (CA); On-site; USD 143,000 - 286,000; Full time

Benefits offered by this job

Stock options

Education benefits

Paid time off

Job summary

Walmart is hiring a Senior Machine Learning Engineer to lead the design, development, and deployment of scalable ML solutions for eCommerce search and discovery. You will collaborate with cross-functional teams to deliver robust models, implement ML pipelines, and ensure compliance and security across systems.

The role focuses on model evaluation, MLOps practices, and continuous improvement while mentoring peers and aligning solutions with corporate goals.

Qualifications

• Degree in computer science, engineering, or related field with relevant ML experience.
• 5+ years of software or ML engineering experience in production systems.
• Experience with MLOps, CI/CD, monitoring, and retraining workflows.

Responsibilities

• Lead design, development, and deployment of scalable ML features for search and discovery.
• Collaborate with engineers, data scientists, and product teams to meet business goals.
• Build robust data pipelines, APIs, and model components ensuring quality and security.
• Apply MLOps practices including versioning, monitoring, and automated retraining.
• Conduct model evaluation, validation, and performance monitoring for reliability.
• Mentor peers and participate in architectural discussions for scalability.

Skills

ML model design

Python

TensorFlow

PyTorch

Scikit-learn

MLOps

SQL

Java/C++/R

GPU optimization

Education

Bachelor's in CS/CE/SE

Master’s degree (preferred)

Tools

TensorFlow

PyTorch

Scikit-learn

Git

Job description

Position Summary...

What you’ll do...

Role summary:

The Senior Machine Learning Engineer will lead the design, development, and deployment of scalable machine learning solutions that address complex business challenges. This role involves collaborating closely with cross-functional teams to deliver robust models, ensuring high-quality code, and maintaining reliable ML pipelines. The position requires expertise in model evaluation, MLOps practices, and secure AI engineering. The successful candidate will drive continuous improvement, mentor peers, and uphold technical excellence while aligning solutions with organizational goals and compliance standards. This role demands strong problem-solving skills and a commitment to innovation and operational resilience.

About the team:

The eCommerce Search team manages the full technology stack enabling product search and discovery across Walmart’s global eCommerce platforms. Supported by microservices, extensive data and feature pipelines, search engines, and ML-serving infrastructure, the team processes millions of daily queries. Key focus areas include query and intent understanding using NLP and deep learning, real-time autocomplete, complex retrieval methods, multi-phase ML ranking models, and data pipelines that feed search indices and analytics. This team’s work directly influences how millions of customers locate products efficiently and effectively. Our systems tackle:

• Query & Intent Understanding - product type classification, intent recognition, and sequence tagging (NLP and deep learning)
• Autocomplete — real-time query suggestions at low-latency
• Retrieval — Complex query construction, faceted navigation, semantic and lexical retrieval
• Ranking — multi-phase ML models (LTR/Neural) optimizing for customer satisfaction
• Data & Feature Pipelines — feeding search indices, feature stores, and analytics platforms.
• What you’ll do:
• Collaborate with engineers, data scientists, and product teams to deliver machine learning features and models aligned with business objectives.
• Design, develop, and deploy scalable ML systems using appropriate algorithms, frameworks, and coding languages.
• Implement and maintain data pipelines, APIs, and model components ensuring quality, security, and compliance.
• Apply MLOps practices including CI/CD, model versioning, monitoring, and automated retraining to sustain performance.
• Conduct rigorous model evaluation, validation, and performance monitoring to ensure accuracy and reliability.
• Mentor team members on best practices, code quality, and model evaluation techniques.
• Participate in architectural discussions to enhance system scalability and maintainability.
• Build and optimize high-throughput, low-latency services, applying best practices around distributed systems, fault-tolerance, and concurrency
• Run A/B experiments; measuring impact on search quality and business metrics

What you’ll bring:

• Proven expertise in designing, developing, and deploying scalable machine learning models using frameworks such as TensorFlow, PyTorch, and Scikit-learn.
• Strong programming skills in Python, SQL, Java, C++, or R, with experience writing clean, production-ready code.
• Proficiency in MLOps practices including CI/CD pipelines, model versioning, automated retraining, and monitoring for performance and drift.
• Experience with hyperparameter optimization, model evaluation, and feature engineering to ensure robust ML solutions.
• Ability to collaborate effectively with cross-functional teams and mentor junior engineers.
• Knowledge of secure, compliant AI engineering and data pipeline development.
• Familiarity with large language modeling and GPU optimization techniques.

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 (https://one.walmart.com/).

The annual salary range for this position is $143,000.00 - $286,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 3 years’ experience in software engineering, machine learning engineering, AI systems or related area.

Option 2: 5 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 1 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 America

Walmart 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, Inc. is an Equal Opportunity Employer- By Choice. We believe we are best equipped to help our associates, customers, and the communities we serve live better when we really know them. That means understanding, respecting, and valuing diversity- unique styles, experiences, identities, abilities, ideas and opinions- while being inclusive of all people.