Staff Machine Learning Engineer - Scale AI Systems & Teams

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

Benefits offered by this job

Stock purchase plan

Health benefits

Paid time off + education benefits

Job summary

Walmart is seeking a Staff Machine Learning Engineer in Sunnyvale to architect and lead scalable, production‑grade ML systems addressing complex business challenges. You will design robust architectures, mentor teams, and drive full ML lifecycle from data sourcing to deployment and monitoring.

You will guide CV solutions using PyTorch, OpenCV, and YOLO, lead cross‑functional collaboration, and champion MLOps practices, including CI/CD and model versioning.

Qualifications

• Extensive experience architecting scalable, production‑grade ML systems and platforms.
• Hands‑on experience designing, developing, and deploying computer vision solutions (object detection, multi‑object tracking, action recognition, video analytics).
• Proven ability to lead technical strategy, mentor engineers, and influence cross‑functional collaboration.

Responsibilities

• Architect and guide scalable, production‑grade ML systems integrated into customer/associate facing products.
• Lead end‑to‑end ML lifecycle execution from data sourcing to deployment and monitoring.
• Partner with cross‑functional teams to align ML solutions with platform strategies.
• Design and build scalable infrastructure for deploying CV models, using TensorRT, GPUs, and containerization.
• Drive performance benchmarking, capacity planning, and cost optimization of ML workloads; establish MLOps best practices.
• Champion CI/CD pipelines, model versioning, drift detection, and telemetry.
• Mentor engineers, shape architectural decisions, and promote engineering excellence.
• Evaluate new patterns and platform capabilities for a forward‑looking ML roadmap.

Skills

ML system design

Production ML

Computer vision

PyTorch

OpenCV

YOLO

GPU inference

MLOps

Mentoring

Technical leadership

Education

Bachelor's degree in CS/CE/ISE/SE or related

Master’s degree in CS or related (preferred)

Tools

TensorRT

NVIDIA GPUs

Docker

Kubernetes

CI/CD

Job description

Walmart is seeking a Staff Machine Learning Engineer in Sunnyvale to architect and lead scalable, production‑grade ML systems addressing complex business challenges. You will design robust architectures, mentor teams, and drive full ML lifecycle from data sourcing to deployment and monitoring.

You will guide CV solutions using PyTorch, OpenCV, and YOLO, lead cross‑functional collaboration, and champion MLOps practices, including CI/CD and model versioning.