Software Engineer AI

Position Summary

Walmart is a multi‑national people‑led tech‑powered omni channel retailer with over 2.2M associates. AtPeople.

AI, we help, support and empower the HR department with AI and Gen AI solutions to improve productivity, harness the opportunities for growth, focus on well‑being and cultivate cultures of belonging to all Walmart's associates.

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 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 Tech's 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.

• Lead/Participate the design, development, and implementation of complex machine learning solutions that drive business results.
• Uphold ML Engineering best practices – enforcing high standards for quality, reliability, and security in deployed machine learning solutions.
• Work cross‑functionally with business, product, data science, front‑end engineering, and data engineering teams to ensure alignment and achieve business goals.
• Effectively communicate technical work to senior/executive leaders in both technical and non‑technical settings.
• Stay up‑to‑date with the latest AI/ML technologies and trend – acting as a thought leader for ML Engineering within the organization and in the broader technical community.
• Mentor junior team members by providing guidance and support to achieve their full potential and improve team efficiency.

• 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/codebase design and ability to translate product requirements into engineering blueprint.
• Experience with common ML model serving framework, especially for NLP models/applications. Familiarity with Gen AI/LLMs and related tooling ecosystem a plus.
• Experience developing and maintaining ML engineering processes and best practices, including code reviews, automated testing, performance optimization, and model monitoring.
• Exceptional leadership and interpersonal skills, with the ability to synthesize requirements across multiple domains and influence stakeholders/peer teams for strategic alignment.
• Strong problem‑solving and analytical skills, with an ability to approach problems in a logical and systematic manner.

Imagine working in an environment where one line of code can make life easier for hundreds of millions of people. That's what we do at Walmart Global Tech. We’re a team of software engineers, data scientists, cybersecurity experts and service professionals within the world's leading retailer who make an epic impact and are at the forefront of the next retail disruption.

People are why we innovate, and people power our innovations. We are people‑led and tech‑empowered. We train our team in the skillsets of the future and bring in experts like you to help us grow. We have roles for those chasing their first opportunity as well as those looking for the opportunity that will define their career. Here, you can kickstart a great career in tech, gain new skills and experience for virtually every industry, or leverage your expertise to innovate at scale, impact millions and reimagine…