Walmart

(USA) Senior, Machine Learning Engineer

Walmart
Apply
4 days ago
Sunnyvale, CA, USASenior
H1B sponsor

Base Salary

$143k - $286k/yr

Responsibilities

  • Collaborate with engineers, data scientists, and product teams to deliver machine learning features and models aligned with business objectives.
  • Design, develop, and deploy scalable machine learning systems using appropriate algorithms, frameworks, and programming languages.
  • Implement and maintain data pipelines, APIs, and model components with attention to quality, security, and compliance.
  • Apply MLOps practices including CI/CD pipelines, model versioning, automated retraining, and performance monitoring.
  • Conduct model evaluation, validation, and performance monitoring to ensure accuracy and reliability.
  • Mentor team members on coding standards, best practices, and model evaluation techniques.
  • Participate in architectural discussions to improve scalability and maintainability.
  • Build and optimize high-throughput, low-latency services using distributed systems, fault-tolerance, and concurrency practices.
  • Run A/B experiments and measure effects on search quality and business metrics.

Requirements

  • Bachelor’s degree in computer science, computer engineering, computer information systems, software engineering, or a related area plus 3 years of experience in software engineering, machine learning engineering, AI systems, or a related area; alternatively, 5 years of relevant experience without the stated degree.
  • Expertise designing, developing, and deploying scalable machine learning models using TensorFlow, PyTorch, and Scikit-learn.
  • Strong programming skills in Python, SQL, Java, C++, or R, with experience writing production-ready code.
  • Proficiency in MLOps practices including CI/CD, model versioning, automated retraining, and monitoring for performance and drift.
  • Experience with hyperparameter optimization, model evaluation, and feature engineering.
  • Ability to collaborate cross-functionally and mentor junior engineers.
  • Knowledge of secure, compliant AI engineering and data pipeline development.
  • Familiarity with large language modeling and GPU optimization techniques.
  • Preferred qualifications include a master’s degree with 1 year of relevant experience, accessibility best practices, inclusive digital experiences, WCAG 2.2 AA standards, and assistive technologies.

Benefits

  • Medical, vision, and dental coverage.
  • 401(k), stock purchase, company-paid life insurance, and performance-based bonus awards.
  • Paid time off and/or PPTO, including sick leave, holidays, 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.
  • Full-time position based at 815 Eleventh Ave, Sunnyvale, California; benefit eligibility may depend on job classification and length of employment.

Tech Stack

C++JavaPythonPyTorchRscikit-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