Walmart

(USA) Staff, Machine Learning Engineer

Walmart
Apply
3 days ago
Bentonville, AR, USA or Sunnyvale, CA, USAStaff+
H1B sponsor

Base Salary

$130k - $338k/yr

Responsibilities

  • Build end-to-end ML lifecycle systems covering data sourcing, feature engineering, model training, deployment, and monitoring.
  • Architect scalable, production-grade ML systems for customer-facing, associate-facing, and operational products.
  • Translate complex business challenges into technical strategies, platform requirements, and success metrics.
  • Implement MLOps practices including model versioning, drift detection, and performance telemetry.
  • Drive innovation in deep learning, NLP, reinforcement learning, and multimodal modeling.
  • Build agentic architecture components including memory, trace, evaluation, orchestration, and routing layers.
  • Mentor engineers and influence architecture and engineering excellence across teams.
  • Collaborate cross-functionally to align ML solutions with platform strategies.

Requirements

  • Bachelor’s degree in computer science, computer engineering, computer information systems, software engineering, or a related area and 4 years of experience in software engineering, machine learning engineering, AI systems, or a related area; alternatively, 6 years of relevant experience.
  • Extensive experience architecting scalable, production-grade machine learning systems and platforms.
  • Proficiency with TensorFlow, PyTorch, Scikit-learn, Python, SQL, and Java.
  • Deep knowledge of neural networks, deep learning, natural language processing, reinforcement learning, and time series analysis.
  • Experience with feature engineering, model training, deployment, monitoring, hyperparameter optimization, model validation, and MLOps practices.
  • Preferred qualifications include a relevant master’s degree with 2 years of related experience, finance-domain experience, computer vision experience, and accessibility experience including WCAG 2.2 AA and assistive technologies.
  • Strong critical-thinking and technical-writing skills.

Benefits

  • Medical, vision, and dental coverage.
  • 401(k), stock purchase program, and company-paid life insurance.
  • Paid time off including sick leave, parental leave, family care leave, bereavement, jury duty, and voting leave.
  • Short-term and long-term disability, company discounts, Military Leave Pay, and adoption and surrogacy expense reimbursement.
  • Walmart-paid education benefits through Live Better U, including tuition, books, and fees for eligible programs from high school completion through bachelor’s degrees and certificates.
  • Additional compensation may include annual or quarterly performance bonuses and stock.
  • The role is based in Sunnyvale, California or Bentonville, Arkansas, with location-dependent annual salary ranges.

Tech Stack

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