Staff ML Engineer: Lead Production-Grade ML Systems

Walmart; Sunnyvale (CA); On-site; USD 169,000 - 338,000; Full time

Benefits offered by this job

Health benefits

401(k) plan

Stock purchase

Paid time off

Live Better U

Job summary

Walmart is seeking a Staff Machine Learning Engineer in Sunnyvale to lead the design and delivery of scalable, production-grade ML solutions. You will oversee data sourcing, model development, deployment, and monitoring while championing MLOps standards.

You'll mentor engineers and collaborate across Product, Data, and Platform teams to push innovation in NLP, CV, and deep learning. The role emphasizes reliability, scalability, and impact on business outcomes.

Qualifications

• Bachelor's degree in CS/engineering or related field and 4 years’ experience in software/machine learning.
• 6 years’ experience in software/ML or related area.

Responsibilities

• Lead the design and deployment of scalable, production-grade ML solutions.
• Translate complex problems into technical requirements and designs.
• Oversee the full ML lifecycle: data sourcing, feature engineering, training, deployment, monitoring.
• Implement MLOps best practices including CI/CD and model versioning.
• Build reusable ML infrastructure components and manage large-scale cloud operations.
• Provide technical leadership and mentorship to engineers.
• Drive experimentation pipelines and explore advanced ML techniques.

Skills

TensorFlow

PyTorch

Scikit-learn

Python

SQL

MLOps

Cloud platforms

Kubernetes

CI/CD

Education

Bachelor's degree

4+ years software/machine learning experience

6 years software/ML experience

Tools

Docker

Kubernetes

CI/CD pipelines

Telemetry/Monitoring

Job description

Walmart is seeking a Staff Machine Learning Engineer in Sunnyvale to lead the design and delivery of scalable, production-grade ML solutions. You will oversee data sourcing, model development, deployment, and monitoring while championing MLOps standards.

You'll mentor engineers and collaborate across Product, Data, and Platform teams to push innovation in NLP, CV, and deep learning. The role emphasizes reliability, scalability, and impact on business outcomes.