Staff, Software Engineer - People.AI

Position Overview

Walmart is a multi-national, people-led, tech-powered omni channel retailer with over 2.2M associates. At People.AI, we support and empower HR with AI and Gen AI solutions to improve productivity, foster growth, and cultivate a culture of belonging for all associates. People.AI is seeking a highly skilled and experienced Staff Machine Learning Engineer to drive the development of innovative Machine Learning and Generative AI systems, architect enterprise-grade ML platforms, microservices, and pipelines. You will be joining the Enterprise People Technology team, which plays a pivotal role in deploying and adopting cutting-edge People technology across Walmart Global Tech’s Enterprise Business Services.

Key Responsibilities

• Lead/participate in the design, development, and implementation of complex machine learning solutions that drive business results.
• Uphold ML Engineering best practices by enforcing high standards for quality, reliability, and security in deployed solutions.
• Collaborate cross-functionally with business, product, data science, front-end engineering, and data engineering teams.
• Effectively communicate technical work to senior and executive leaders in both technical and non-technical terms.
• Stay current with the latest AI/ML technologies and trends, acting as a thought leader within the organization and broader community.
• Mentor junior team members to help them achieve their full potential and improve overall team efficiency.

Required Qualifications

• Deep knowledge and experience in building complex, large-scale, mission-critical machine learning applications with microservice architecture in Kubernetes.
• Proven programming skills using standard ML tools such as Python, TensorFlow, PyTorch, CUDA, and related frameworks and libraries.
• Highly resourceful and methodical in solution, architecture, API, and codebase design, with the ability to translate product requirements into engineering blueprints.
• Experience with ML model serving frameworks, particularly for NLP models/applications. Familiarity with Gen AI/LLMs is a plus.
• Experience developing and maintaining ML engineering processes such as code reviews, automated testing, performance optimization, and model monitoring.
• Exceptional leadership and interpersonal skills with the capacity to synthesize requirements across multiple domains and influence stakeholders.
• Strong problem-solving and analytical skills with a logical and systematic approach.
• Option 1: Education: Bachelor's degree in computer science, computer engineering, computer information systems, software engineering, or a related area and 4 years' experience in software engineering or a related field.
• Option 2: 6 years' experience in software engineering or a related field.

Preferred Qualifications

• Master's degree in Computer Science, Computer Engineering, Computer Information Systems, Software Engineering, or a related area and 2 years' experience in software engineering or a related field.
• Background in creating inclusive digital experiences, including knowledge of Web Content Accessibility Guidelines (WCAG) 2.2 AA standards and assistive technologies.
• Familiarity with accessibility best practices to help build accessible products and services in line with Walmart's standards and guidelines.

Benefits & Perks

• Salary: Bentonville, Arkansas US-09430 & Dallas, Texas US-09648: $110,000.00 - $220,000.00; Bellevue, Washington US-11075: $132,000.00 - $264,000.00.
• Compensation: Includes annual or quarterly performance bonuses and, for certain positions, stock awards.
• Benefits: Comprehensive health benefits (medical, vision, dental), financial benefits including 401(k), stock purchase, company-paid life insurance, and generous paid time off (PTO, sick leave, parental leave, and more). Additionally, the Live Better U education benefit program is available.
• Location: 640 W California Avenue, Sunnyvale, CA 94086-4828, United States of America.