Walmart logo

Staff, Software Engineer - People.AI

Walmart

Bentonville, AR
Full Time
Senior
132k-264k
15 days ago

Job Description

About the Role

People.AI is seeking a highly skilled and experienced Staff Machine Learning Engineer. You will be responsible for leading the development of innovative Machine Learning and Generative AI systems, as well as architecting enterprise-grade Machine Learning platform, microservices, and pipelines. The team supports the deployment and adoption of People technology across Walmart, impacting millions of associates globally. The role involves building scalable AI solutions to improve HR productivity, workforce planning, and fostering a culture of well-being and inclusion for Walmart associates.

Key Responsibilities

  • Lead the design and implementation of enterprise knowledge graphs, including ontology design, schema modeling, and graph-based reasoning.
  • Build robust data pipelines for knowledge graph ingestion using Python, Airflow, and PySpark.
  • Manage and optimize knowledge graph databases, including schema design, performance tuning, and query optimization using Cypher.
  • Develop and maintain GraphQL APIs for exposing graph data to downstream services and end-user applications.
  • Collaborate cross-functionally with data scientists, ML engineers, HR partners, and product teams to translate business needs into technical solutions.
  • Work with semantic search, entity linking, and graph-based querying to power intelligent applications and AI assistants.
  • Enforce best practices in data governance, scalability, observability, and security in all graph-related systems.

Requirements

  • Proven experience with knowledge graph databases, including designing and managing large-scale graph databases in production.
  • Strong understanding of knowledge graph construction, entity resolution, and schema alignment techniques.
  • Proficiency in Python, with experience in PySpark and Airflow for ETL and data orchestration.
  • Strong command of Cypher and hands-on experience with GraphQL API development.
  • Background in data modeling and working with complex data relationships across multiple domains.
  • Experience working in ML- or AI-driven environments, particularly involving semantic data, LLMs, or intelligent assistants, is a plus.
  • Excellent collaboration and communication skills, including the ability to work with technical and non-technical stakeholders.
  • Bachelor's or Master's degree in Computer Science, Data Engineering, or a related field.

Nice to Have

  • Experience in creating inclusive digital experiences and knowledge of Web Content Accessibility Guidelines (WCAG) 2.2 AA standards.
  • Knowledge of assistive technologies and integrating digital accessibility to support an inclusive culture.

Qualifications

  • Option 1: Bachelor's degree in computer science, computer engineering, computer information systems, software engineering, or related area and 4 years' experience in software engineering or related area.
  • Option 2: 6 years' experience in software engineering or related area.
  • Preferred: Master's degree in Computer Science, Computer Engineering, Computer Information Systems, Software Engineering, or related area and 2 years' experience in software engineering or related area.

Benefits & Perks

  • Competitive pay with performance-based bonus awards.
  • Health benefits including medical, vision, and dental coverage.
  • Financial benefits such as 401(k), stock purchase, and company-paid life insurance.
  • Paid time off including PTO, sick leave, parental leave, family care leave, bereavement, jury duty, and voting.
  • Other benefits including short-term and long-term disability, company discounts, Military Leave Pay, adoption and surrogacy expense reimbursement.
  • PTO and PPTO for vacation, sick leave, holidays, or other purposes.
  • Live Better U education benefit program covering tuition, books, and fees.

Working at Walmart

The Walmart Global Tech team supports the successful deployment and adoption of People-focused technologies across Walmart. The company is committed to building intelligent systems that make HR operations smarter and more responsive, fostering an inclusive culture and creating accessible products and services following Walmart's accessibility standards and guidelines.

Apply Now

Job Details

Posted AtJul 9, 2025
Salary132k-264k
Job TypeFull Time
ExperienceSenior

Job Skills

AI Insights

Key skills identified from this job posting

Sign upto access all insights for this job

About Walmart

Website

walmart.com

Company Size

10000+ employees

Location

Bentonville, AR

Industry

Warehouse Clubs, Supercenters, and Other General Merchandise Retailers

Get job alerts

Set up personalized alerts for your job search and get tailored job digests for close matches