Walmart

Staff Machine Learning Engineer

Walmart

full-time

Posted on:

Location Type: Hybrid

Location: BentonvilleCaliforniaUnited States

Visit company website

Explore more

AI Apply
Apply

Salary

💰 $130,000 - $260,000 per year

Job Level

About the role

  • Develop AI driven intelligent products for the Associate Productivity and Experience team
  • Design, architect, build AI/ML models, AI Agents and deploy, operate, optimize the solutions
  • Work on Python and mainstream machine learning frameworks, e.g. TensorFlow or PyTorch and Agentic Frameworks like Pydantic
  • Lead development of end-to-end system architecture for GenAI/AI/ML and data intensive applications
  • Design, Develop and deploy Full stack based applications.
  • Develop and deploy production-grade real-time/batch machine learning services
  • Work with product managers to design user journeys, feedback loop and analyze user telemetry
  • Identify or propose AI/ML use-cases to business teams to improve business processes and develop quick MVPs/ POCs
  • End to end ownership of product using technologies such as Java, Node JS, React Js
  • Review design and code, provide technical solutions to team for complex requirements
  • Uphold and enforce rigorous standards, governance, and best practices, ensuring compliance with model and data governance standards.

Requirements

  • bachelor's/master's degree in computer science, information technology, engineering, or related area
  • Minimum 10 years of experience in design, development, deployment, and maintenance of highly scalable, robust, and user-centric Fullstack/GenAI /AI /ML products and applications
  • Good understanding of Data Structures
  • Hands on experience in systems design, algorithms, and distributed systems.
  • Industry level experience in developing production-grade AI/ML system development with major technology companies or AI-native start-ups.
  • Experience at writing high-performing production-quality code in Python
  • Exposure to cloud infrastructure, such as Open Stack, Azure, GCP, or AWS
  • Experience working in Agile Methodology
  • Experience using CI/CD, tools for logging and metrics
  • A continuous drive to explore, improve, enhance, automate and optimize systems and tools.
  • Strong computer science fundamentals in data structures and algorithms
  • Exposure to information retrieval, statistics, and machine learning.
  • Excellent oral and written communication skills.
  • Good analytical and problem-solving skills.
Benefits
  • Health benefits include medical, vision and dental coverage.
  • 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.
  • Tuition, books, and fees are completely paid for by Walmart for education programs.
Applicant Tracking System Keywords

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

Hard Skills & Tools
PythonAI/ML modelsTensorFlowPyTorchPydanticFull stack developmentJavaNode JSReact JSData Structures
Soft Skills
communication skillsanalytical skillsproblem-solving skillsleadershipcollaboration
Certifications
bachelor's degreemaster's degree