Walmart

(USA) Senior, Machine Learning Engineer

Walmart
Apply
1 day ago
Sunnyvale, CA, USASenior
H1B sponsor

Base Salary

$143k - $286k/yr

Responsibilities

  • Design and develop auto-research capabilities that automate model training, optimization, experimentation, evaluation, and hyperparameter tuning.
  • Build, deploy, test, and support machine learning services, APIs, and data pipelines.
  • Apply MLOps practices including model versioning, automated retraining, and CI/CD.
  • Monitor and maintain models to address drift, performance issues, quality, and reliability.
  • Ensure machine learning pipelines meet security, compliance, and enterprise AI governance standards.
  • Collaborate with engineers, data scientists, and product teams while providing technical leadership and mentoring peers.

Requirements

  • Bachelor’s degree in computer science, computer engineering, computer information systems, software engineering, or a related area plus three years of relevant experience, or five years of relevant experience without the degree.
  • Expertise designing, developing, and deploying scalable machine learning systems using TensorFlow, PyTorch, and Scikit-learn.
  • Strong proficiency in Python, SQL, Java, and C++.
  • Hands-on experience operating agentic or autonomous research systems and running automated experimentation 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 model versioning, automated retraining, telemetry, and CI/CD.
  • Ability to implement secure and compliant ML pipelines aligned with enterprise AI governance standards.
  • Master’s degree plus one year of relevant experience is preferred.
  • Experience with accessibility best practices, inclusive digital experiences, WCAG 2.2 AA standards, and assistive technologies is preferred.

Benefits

  • Medical, vision, and dental coverage.
  • 401(k), stock purchase benefits, company-paid life insurance, and potential stock awards.
  • Paid time off including sick leave, parental leave, family care leave, bereavement, jury duty, voting leave, and vacation or holiday use through PTO/PPTO.
  • Short-term and long-term disability, company discounts, Military Leave Pay, and adoption and surrogacy expense reimbursement.
  • Walmart-paid education benefits covering eligible tuition, books, and fees for programs ranging from high school completion to bachelor’s degrees and certificates.
  • Annual or quarterly performance bonuses in addition to base salary.

Tech Stack

C++JavaPythonPyTorchscikit-learnSQLTensorFlow

Categories

Walmart

About Walmart

10,000+ employees

Walmart is a global omnichannel retailer that sells groceries, general merchandise, and services through Walmart and Sam's Club stores and online marketplaces for consumers and small businesses. Its model spans physical retail, e-commerce, advertising, and a vast supply chain, plus pharmacy, health, and financial services. Founded in 1962 and headquartered in Bentonville, Arkansas, Walmart Inc. is a public company on the NYSE (WMT) operating in the U.S. and multiple international markets.

Contact me