Senior ML Engineer — Auto-Research & Production AI

Walmart Inc.; Sunnyvale (CA); On-site; USD 143,000 - 286,000; Full time

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

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.