Senior, Software Engineer - MLEWalmart • Sunnyvale, CA, United States

Senior, Software Engineer - MLE

Walmart • Sunnyvale, CA, United States

job_description.job_card.variable_days_ago

serp_jobs.job_preview.job_type

serp_jobs.job_card.full_time

job_description.job_card.job_description

Position Summary

We are seeking a highly motivated Machine Learning Engineer to join our Data Science team. In this role you will design, develop, and deploy machine learning models at scale and play a key role in shaping next‑generation AI / ML and GenAI enabled products that help our suppliers grow their business.

What you’ll do

Build and maintain end‑to‑end ML infrastructure for data ingestion, feature engineering, training, deployment, and monitoring.
Develop tools and frameworks that streamline experimentation for data scientists.
Ensure best practices in MLOps : CI / CD for ML, automated retraining, model versioning, and monitoring.
Optimize system performance, reliability, and cost efficiency in production environments.
Partner with data scientists to move research prototypes into robust, scalable systems.
Collaborate with platform and infrastructure engineers to optimize compute, storage, and deployment efficiency.
Advocate for ML engineering best practices and scalable design patterns.
Mentor junior engineers and data scientists on production readiness and infrastructure use.
Enable design, train, and deploy ML models or recommendation systems (ranking, retrieval, embeddings, personalization).
Integrate recommendation outputs into a centralized agentic AI system to enhance reasoning and decision‑making.
Support workflows such as planning, contextual decision‑making, and multi‑agent coordination.

What You Will Bring

Bachelor’s, Master’s, or Ph.D. in Computer Science, Machine Learning, or a related field.

Strong expertise in MLOps and infrastructure engineering (CI / CD, monitoring, automation, Docker / Kubernetes, cloud ML platforms).

Experience with cloud platforms (AWS, Azure, GCP).

Proficiency in Python and ML frameworks (PyTorch, TensorFlow, Scikit‑learn) and experience with distributed data / compute systems.

Experience building real‑time and batch processed inference systems.

Strong understanding of data structures, algorithms, and statistical modeling.

Experience with data preprocessing, feature engineering, ML training, and model evaluation.

Experience with pipeline optimization (multi‑processing, multi‑threading, distributed frameworks, caching, etc.).

Excellent problem‑solving and analytical skills.

Strong communication and collaboration skills.

Preferred skills

Hands‑on experience building and deploying ML systems at scale.

Familiarity with LLMs, reinforcement learning, or agentic AI frameworks.

Prior experience creating scalable ML platforms.

Benefits

At Walmart, we offer competitive pay, performance‑based bonus awards, and other benefits. Health benefits include medical, vision, and dental coverage. Financial benefits include 401(k), stock purchase, and company‑paid life insurance. Paid time off benefits include PTO, parental leave, family care leave, bereavement, jury duty, and voting. Other benefits include short‑term and long‑term disability, company discounts, military leave pay, adoption and surrogacy expense reimbursement, and more.

Compensation

Sunnyvale, California — Annual salary range : $117,000.00 – $234,000.00

Bentonville, Arkansas — Annual salary range : $90,000.00 – $180,000.00

Additional compensation includes annual or quarterly performance bonuses. Additional compensation for certain positions may also include stock.

#J-18808-Ljbffr

serp_jobs.job_alerts.create_a_job

Senior Software Engineer • Sunnyvale, CA, United States