(USA) Senior, Machine Learning Engineer

Walmart Inc.

Sunnyvale (CA)

On-site

USD 143,000 - 286,000

Full time

6 hours ago
Be an early applicant

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Benefits offered by this job

Health benefits
401(k)
Stock purchase plan
Life insurance
PTO / Paid time off
Live Better U tuition

Job summary

Walmart Inc. in Sunnyvale, CA, is seeking a Senior Machine Learning Engineer to lead the design, development, and deployment of scalable ML solutions.

You will build auto-research capabilities to automate training, experimentation, and evaluation, enabling continuous improvements across the ML lifecycle. You will collaborate with cross-functional teams to deliver robust, production-ready models and data pipelines while ensuring security and governance standards.

Qualifications

  • Bachelor's degree in computer science, computer engineering, or related field with 3 years of software/ML engineering experience; or 5 years of related experience.
  • Experience designing, developing, and deploying scalable ML systems using TensorFlow, PyTorch, and Scikit-learn.
  • Proficiency in Python, SQL, Java, and C++ for production ML applications.

Responsibilities

  • Collaborate with engineers, data scientists, and product teams to deliver ML features and models.
  • Design and develop auto-research capabilities to automate training, optimization, experimentation, and evaluation.
  • Surface top-performing models with minimal manual effort to accelerate the ML lifecycle.
  • Implement, test, and support ML services, APIs, and data pipelines for reliability and quality.
  • Apply MLOps practices including CI/CD, model versioning, and automated retraining.

Skills

TensorFlow
PyTorch
Scikit-learn
Python
SQL
Java
C++
Autonomous research
AutoResearch loops
ML lifecycle
MLOps
CI/CD
Model versioning
Telemetry
Security & governance
Mentorship

Education

Bachelor's degree in Computer Science
Master's degree in Computer Science

Job description

Position Summary...
TheSeniorMachineLearningEngineerwillleadthedesign,development,anddeploymentofscalablemachinelearningsolutionsthataddresscomplexbusinesschallenges,withaprimaryfocusonbuildingauto-researchcapabilitiesthatautomatemodeltrainingoptimization,experimentation,andevaluationtoacceleratetheMLdevelopmentlifecycle.Thisrolerequirescollaborationwithcross-functionalteamstodeliverrobust,production-readymodelsanddatapipelineswhileensuringcompliancewithsecurityandgovernancestandards.Additionally,theywillprovidetechnicalleadership,mentorpeers,anddrivecontinuousimprovementtoenhancesystemreliabilityandinnovationalignedwithorganizationalgoals.

About the team:

Our team develops foundational models tailored for Walmart’s e-commerce platform by leveraging large-scale user interactions to create rich representations of users, products, and search queries. These models enable scalable search, advertising, recommendation, and personalization solutions. Team members apply expertise in data science, machine learning, and software engineering to optimize model performance, training, and deployment. With a strong focus on data quality, advanced analytics, and impactful visualization, the team delivers actionable insights that influence key business outcomes. Cross-functional collaboration drives innovation and continuous improvement in a dynamic technology environment.

What you'll do...
  • Collaborate with engineers, data scientists, and product teams to deliver machine learning features and models.
  • Design and develop auto-research capabilities that automate model training, optimization, experimentation, and evaluation, enabling AI agents to iteratively tune hyperparameters and run experiments against defined objective metrics.
  • Surface the best-performing models with minimal manual intervention, accelerating the overall machine learning development lifecycle.
  • Implement, test, and support machine learning services, APIs, and data pipelines, ensuring quality and reliability.
  • Apply MLOps practices, including CI/CD, model versioning, and automated retraining, to sustain performance.
  • Conduct rigorous model evaluation, monitoring, and maintenance to address model drift and performance issues.
  • Ensure secure, compliant machine learning engineering aligned with enterprise AI governance standards.

What you'll bring:

  • Proven expertise in designing, developing, and deploying scalable machine learning systems using frameworks such as TensorFlow, PyTorch, and Scikit-learn.
  • Strong proficiency in programming languages including Python, SQL, Java, and C++ to build production-ready machine learning applications.
  • Hands-on experience operating agentic or autonomous research systems.
  • Experience running automated experimentation against defined objective metrics through AutoResearch loops.
  • Experience with machine learning lifecycle management, including data sourcing, feature engineering, model training, evaluation, deployment, and monitoring.
  • Knowledge of MLOps practices such as CI/CD, model versioning, automated retraining, and telemetry to ensure system reliability.
  • Ability to implement secure, compliant machine learning pipelines that adhere to enterprise AI governance standards.
  • Demonstrated collaboration and mentorship capabilities within cross-functional teams to drive technical excellence.

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 $143,000.00 - $286,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 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 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 94089-4731

Distinguished, Software Engineer - Machine Learning

Sunnyvale, CA + 1 more

(USA) Senior, Machine Learning Engineer

Sunnyvale, CA 94089-4731

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

(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, Machine Learning Engineer
(USA) Principal, Machine Learning Engineer

Walmart Inc. • Sunnyvale (CA)

On-site
USD 169,000 - 338,000
Health benefits
401(k)
Stock purchase
+2
(USA) Staff, Machine Learning Engineer
(USA) Staff, Machine Learning Engineer

Walmart Inc. • Sunnyvale (CA)

On-site
USD 169,000 - 338,000
Discount Card
Live Better U (tuition paid)
Medical, dental, vision
+3
(USA) Senior, Data Scientist
(USA) Senior, Data Scientist

Walmart Inc. • Sunnyvale (CA)

On-site
USD 117,000 - 234,000
Discount Card
Live Better U
Walmart Academy
+2
Senior AI & Cloud Architect Engineer
Senior AI & Cloud Architect Engineer

Walmart Inc. • Sunnyvale (CA)

On-site
USD 169,000 - 338,000
Discount Card
Live Better U
Medical benefits
+2
(USA) Senior, Data Scientist
(USA) Senior, Data Scientist

Walmart Inc. • Bentonville (AR)

On-site
USD 90,000 - 180,000
Medical, dental, vision
401(k) and stock purchase
Paid time off
+2
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, 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
Senior Manager, Software Platform & DevOps Engineering
Senior Manager, Software Platform & DevOps Engineering

Walmart Inc. • Sunnyvale (CA)

On-site
USD 143,000 - 286,000
Competitive pay
Performance-based bonus
Medical, vision and dental
+3
(USA) Staff, Software Engineer
(USA) Staff, Software Engineer

Walmart Inc. • Bellevue (WA)

On-site
USD 132,000 - 264,000
Health insurance
Live Better U education program
Stock purchase plan