Walmart

Associate Fellow I, Machine Learning Engineer

Walmart

full-time

Posted on:

Location Type: Office

Location: SunnyvaleCaliforniaUnited States

Visit company website

Explore more

AI Apply
Apply

Salary

💰 $254,000 - $481,000 per year

About the role

  • The Associate Fellow I is a highly technical, hands-on role within the Walmart Global Tech team, responsible for supporting architectural designs and advisory across multiple enterprise business domain areas.
  • Lead and inspire high-performing engineering teams to build and scale highly reliable, secure, and performant platform/products and frameworks.
  • Drive a results-oriented culture with a strong focus on engineering excellence, execution and propagate build for scale mindset.
  • Develop trusted partnerships with developers, business partners, product, and architecture leaders to drive optimized platform/product delivery.
  • Lead complex technical and strategic discussions that involve multiple personas including engineering, architect, product, associate, and other stakeholders.
  • Assess and leverage technology when appropriate from Global solutions existing in Walmart ecosystem to build/leverage scalable and innovative solutions.
  • Build, mentor, and teach a community of engineers, and act as a central resource to provide advisory on complex projects and issues.
  • Possess a high sense of urgency to solve problems and executing them with ownership.

Requirements

  • Master’s in Computer Science, ML, or related field (or equivalent).
  • 8+ years building and deploying ML models in production.
  • Hands-on experience with ML frameworks (TensorFlow, PyTorch, Spark ML, XGBoost).
  • Strong coding proficiency in Python, SQL, Java, C++, or R.
  • Demonstrated ability to lead architecture design for domain-specific solutions.
  • Contributions to enterprise-wide AI/ML initiatives or open-source projects.
  • Experience applying responsible AI and compliance practices to production systems.
  • Experience with pipeline orchestration (Airflow, Dagster, Prefect).
  • Familiarity with AutoML, NAS, or advanced feature engineering.
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 (including sick leave), 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.
  • Live Better U is a Walmart-paid education benefit program for full-time and part-time associates in Walmart and Sam's Club facilities. Programs range from high school completion to bachelor's degrees, including English Language Learning and short-form certificates. Tuition, books, and fees are completely paid for by Walmart.
Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools
machine learningML frameworksTensorFlowPyTorchSpark MLXGBoostPythonSQLJavaC++
Soft Skills
leadershipmentoringproblem-solvingcommunicationcollaborationexecutionownershipresults-orientedinspirationpartnership
Certifications
Master’s in Computer ScienceMaster’s in ML