(USA) Senior, Machine Learning Engineer

Walmart • Sunnyvale, CA

2d • $143,000 - $286,000

About The Position

The Senior Machine Learning Engineer leads the design, development, and deployment of scalable machine learning solutions for complex business challenges. The role focuses on building auto-research capabilities that automate model training optimization, experimentation, and evaluation to accelerate the machine learning development lifecycle. The engineer collaborates with cross-functional teams to deliver production-ready models and data pipelines while meeting security and governance standards, and provides technical leadership and mentorship to improve reliability and innovation.

The team develops foundational models for Walmart’s e-commerce platform using large-scale user interactions to represent users, products, and search queries. These models support search, advertising, recommendations, and personalization. The team combines data science, machine learning, and software engineering expertise, with a focus on data quality, model performance, advanced analytics, and collaboration.

Requirements

• Bachelor’s degree in computer science, computer engineering, computer information systems, software engineering, or a related area, plus 3 years of experience in software engineering, machine learning engineering, AI systems, or a related area; or 5 years of experience in one of those areas.
• Expertise designing, developing, and deploying scalable machine learning systems using frameworks such as TensorFlow, PyTorch, and Scikit-learn.
• Proficiency in Python, SQL, Java, and C++ for 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 across the machine learning lifecycle, 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.
• Ability to implement secure, compliant machine learning pipelines aligned with enterprise AI governance standards.
• Collaboration and mentorship capabilities within cross-functional teams.

Nice To Haves

• Master’s degree in computer science, computer engineering, computer information systems, software engineering, or a related area, plus 1 year of experience in software engineering, machine learning engineering, AI systems, or a related area.
• Prior work experience in accessibility best practices.
• Experience creating inclusive digital experiences, including implementing WCAG 2.2 AA standards, using assistive technologies, and integrating digital accessibility.

Responsibilities

• 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, including agent-driven hyperparameter tuning against defined objective metrics.
• Identify the best-performing models with minimal manual intervention to accelerate the 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.
• Evaluate, monitor, and maintain models to address drift and performance issues.
• Ensure machine learning engineering is secure and compliant with enterprise AI governance standards.
• Provide technical leadership and mentor peers.

Benefits

• Medical, vision, and dental coverage.
• 401(k).
• Stock purchase plan.
• Company-paid life insurance.
• Paid time off, including sick leave, parental leave, family care leave, bereavement leave, jury duty, and voting leave.
• Short-term and long-term disability.
• Company discounts.
• Military Leave Pay.
• Adoption and surrogacy expense reimbursement.
• Live Better U education benefit, with tuition, books, and fees paid for eligible programs.
• Performance-based bonus awards.
• Stock may be included as additional compensation for certain positions.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior