Senior, Machine Learning Engineer

Lead the end-to-end delivery of scalable machine learning systems, with a focus on building auto-research capabilities for automated training optimization, experimentation, and evaluation.

Responsibilities

• Partner with engineers, data scientists, and product teams to deliver machine learning features and models.
• Design and build auto-research capabilities that automate training, optimization, experimentation, and evaluation.
• Run iterative model tuning workflows that adjust hyperparameters and evaluate against defined objective metrics.
• Identify and surface top-performing models with minimal manual intervention to speed the ML development lifecycle.
• Implement, test, and support machine learning services, APIs, and data pipelines.
• Use MLOps practices such as CI/CD, model versioning, and automated retraining to sustain model performance.
• Perform rigorous evaluation, monitoring, and ongoing maintenance to manage model drift and performance issues.
• Ensure secure and compliant machine learning engineering that aligns with enterprise AI governance standards.

Requirements

• Proven expertise designing, developing, and deploying scalable machine learning systems using TensorFlow, PyTorch, and Scikit-learn.
• Strong programming skills in Python, SQL, Java, and C++ for production-ready ML applications.
• Hands-on experience operating agentic or autonomous research systems.
• Experience running automated experimentation against defined objective metrics through AutoResearch loops.
• Experience across the ML lifecycle: data sourcing, feature engineering, model training, evaluation, deployment, and monitoring.
• Knowledge of MLOps: CI/CD, model versioning, automated retraining, and telemetry.
• Ability to implement secure, compliant machine learning pipelines aligned with enterprise AI governance standards.
• Demonstrated collaboration and mentorship within cross-functional teams.

Technologies

• TensorFlow
• PyTorch
• Scikit-learn
• Python
• SQL
• Java
• C++
• CI/CD
• AutoResearch
• Telemetry

Team

• Develops foundational models for Walmart’s e-commerce platform using large-scale user interaction data.
• Builds rich representations of users, products, and search queries to support search, advertising, recommendation, and personalization.
• Applies data science, machine learning, and software engineering to optimize model performance, training, and deployment.
• Emphasizes data quality, advanced analytics, and visualization to deliver actionable insights.
• Uses cross-functional collaboration to drive innovation and continuous improvement.

Education

• Bachelor’s degree in computer science, computer engineering, computer information systems, software engineering, or related field (as listed).

Minimum Qualifications

• Option 1: Bachelor’s degree (as listed) and 3 years of experience in software engineering, machine learning engineering, AI systems, or related area.
• Option 2: 5 years of experience in software engineering, machine learning engineering, AI systems, or related area.

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 creating inclusive digital experiences, including knowledge of implementing WCAG 2.2 AA standards, assistive technologies, and integrating digital accessibility.

Location

• Sunnyvale, CA, United States (Onsite)
• 815 Eleventh Ave, Sunnyvale, CA 94089-4731, United States of America

Compensation

• USD 143,000 - 286,000 per year

Benefits

• Health benefits: medical, vision, and dental coverage.
• Financial benefits: 401(k), stock purchase, and company-paid life insurance.
• Paid time off: PTO (including sick leave), parental leave, family care leave, bereavement, jury duty, and voting.
• Other benefits: short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement, and more.
• Live Better U education benefit program for full-time and part-time associates in Walmart and Sam’s Club facilities.
• Tuition, books, and fees are completely paid for by Walmart.