Mlops Machine Learning Engineer Jobs in Milpitas, CA

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 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.