Staff, Software Engineer - People.AI

Staff Machine Learning Engineer

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.

About Team: The Enterprise People Technology team supports the successful deployment and adoption of new People technology across the enterprise. As a Fortune #1 company, our work impacts millions of associates globally. We strive to continuously improve people technology and products to help managers and associates so they can focus on what matters most - supporting our customers and members. People Technology is one of the major segments of Walmart Global Techs Enterprise Business Services, which is invested in building a compact, robust organization that includes service operations and technology solutions for Finance, People, and the Associate Digital Experience.

What You'll Do

At People.AI, we build and scale AI and GenAI solutions to empower the HR functionimproving productivity, enabling strategic workforce planning, and supporting a culture of well-being and inclusion for all Walmart associates.

We are seeking a Staff Graph Engineer with deep expertise in graph database, knowledge graph design, and graph data pipelines. This role will play a key part in building scalable graph-based systems that power insights across Walmart's associate experience, including use cases such as Cross Channel Listening, Workforce Planning, and Organizational Intelligence.

• 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.

What You'll Bring:

• 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.

About the Team

The Walmart Global Tech team supports the successful deployment and adoption of People-focused technologies across Walmart. As a Fortune #1 company, our work impacts millions of associates globally. We are committed to building intelligent systems that make HR operations smarter and more responsiveso managers and associates can focus on what matters most: our customers and communities.

At Walmart, we offer competitive pay as well as performance-based bonus awards and other great benefits for a happier mind, body, and wallet. 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.

You will also receive PTO and/or PPTO that can be used for vacation, sick leave, holidays, or other purposes. The amount you receive depends on your job classification and length of employment. It will meet or exceed the requirements of paid sick leave laws, where applicable.

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. Eligibility requirements apply to some benefits and may depend on your job classification and length of employment. Benefits are subject to change and may be subject to a specific plan or program terms.

For information about benefits and eligibility, see One.Walmart.

Minimum 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 Qualifications

• 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.
• We value candidates with a background in creating inclusive digital experiences, demonstrating knowledge in implementing Web Content Accessibility Guidelines (WCAG) 2.2 AA standards, assistive technologies, and integrating digital accessibility seamlessly.
• The ideal candidate would have knowledge of accessibility best practices and join us as we continue to create accessible products and services following Walmart's accessibility standards and guidelines for supporting an inclusive culture.

Primary Location

508 Sw 8Th St, Bentonville, AR 72712, United States of America

About Walmart

At Walmart, we help people save money so they can live better. This mission serves as the foundation for every decision we make, from responsible sourcing to sustainabilityand everything in between. As a Walmart associate, you will play an integral role in shaping the future of retail, tech, merchandising, finance and hundreds of other industriesall while affecting the lives of millions of customers all over the world. Here, your work makes an impact every day. What are you waiting for? Walmart Inc. is an Equal Opportunity Employer- By Choice. We believe we are best equipped to help our associates, customers, and the communities we serve live better when we really know them. That means understanding, respecting, and valuing unique styles, experiences, identities, abilities, ideas, and opinions- while welcoming all people.