Position Summary...

We are seeking a highly motivated Machine Learning Engineer to join our Data Science team. In this role, you will not only design, develop, and deploy machine learning models at scale but also play a key role in shaping next-generation AI/ML and GenAI enabled products that will help our suppliers grow their business. You will collaborate with data scientists, software engineers, and product stakeholders to ensure ML models and AI agents seamlessly integrate into production systems, driving measurable business impact.

What you'll do...

• Infrastructure & MLOps

• Build and maintain end-to-end ML infrastructure for data ingestion, feature engineering, training, deployment, and monitoring.

• Develop tools and frameworks that streamline experimentation for data scientists.

• Ensure best practices in MLOps: CI/CD for ML, automated retraining, model versioning, and monitoring.

• Optimize system performance, reliability, and cost efficiency in production environments.

• Collaboration & Leadership

• Partner with data scientists to move research prototypes into robust, scalable systems.

• Collaborate with platform and infrastructure engineers to optimize compute, storage, and deployment efficiency.

• Advocate for ML engineering best practices and scalable design patterns.

• Mentor junior engineers and data scientists on production readiness and infrastructure use.

• ML Systems/Recommendation Systems

• Enable design, train, and responsible for deploying ML models or Recommendation System (ranking, retrieval, embeddings, personalization).

• Collaborate with data scientists to bring prototypes into production.
   

• Agentic AI Enablement

• Integrate recommendation outputs into a centralized agentic AI system to enhance reasoning and decision-making.

• Support workflows such as planning, contextual decision-making, and multi-agent coordination.

What You Will Bring

• Bachelor’s/Master’s/Ph.D. in Computer Science, Machine Learning, or a related field.

• Strong expertise in MLOps and infrastructure engineering (CI/CD, monitoring, automation, Docker/Kubernetes, cloud ML platforms).

• Experience with cloud platforms (e.g., AWS, Azure, GCP).

• Proficiency in Python and ML frameworks (PyTorch, TensorFlow, Scikit-learn), plus experience with distributed data/compute systems.

• Experience in building real time and batch processed inference system

• Strong understanding of data structures, algorithms, and statistical modeling.

• Experience with data preprocessing, feature engineering, ml training and model evaluation.

• Experience with pipeline optimization viz. Multi-processing, Multi-threading, Distributed frameworks, Caching , etc.

• Excellent problem-solving and analytical skills.

• Strong communication and collaboration skills.

Preferred skills:

• Hands-on experience building and deploying ML systems at scale.

• Familiarity with LLMs, reinforcement learning, or agentic AI frameworks

• Prior of experience in creating scalable ML platforms

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.

‎ 

Sunnyvale, California US-11657:The annual salary range for this position is $117,000.00-$234,000.00

‎ 

Bentonville, Arkansas US-10735:The annual salary range for this position is $90,000.00-$180,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 or related area.
Option 2: 5 years’ experience in software engineering 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 year's experience in software engineering or related area., We value candidates with a background 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. The ideal candidate would have knowledge of accessibility best practices and join us as we continue to create accessible products and services following Walmart’s accessibility standards and guidelines for supporting an inclusive culture.

Primary Location...

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