(USA) Staff, Software Engineer, Machine Learning

Staff Software Engineer

As a part of Walmart Global Tech, you will build state-of-the-art personalization and recommendation solutions that will elevate the e-commerce experience of millions of customers. We are looking for a Staff Software Engineer to join the Personalization team under the Customer Experience organization. You will be in the unique position to bring a broader impact to Walmart’s customers and associates by building features and solutions that power the ML model inferencing and governance at scale, contributing to highly engaging customer experiences for Walmart customers across channels.

If you are someone who feels a personal stake in everything that you work on, has a love for data, enjoys solving complex problems, has a passion for privacy, knows how to foster strong relationships and build trust, and works for the success of the entire team then this is just the right place for you. Our team consists of big data geeks, application engineers, scientists, and product visionaries who are passionate about using cutting-edge machine learning, deep learning, reinforcement learning, natural language processing solutions with data at the scale of Walmart to deliver personalization and recommendation products.

Our team works closely with our US stores and eCommerce business to better serve customers by empowering team members, stores, and merchants with technological innovation. From groceries and entertainment to sporting goods and crafts, Walmart U.S. offers an extensive selection that our customers value, whether they shop online at Walmart.com, through one of our mobile apps, or in-store. Focus areas include customers, stores and employees, in-store service, merchant tools, merchant data science, and search and personalization.

What you will do:

• Lead development of multi-layered feature store stacks supporting hybrid retrieval, diverse data modalities, and multi-tenant access control that powers modern Agentic flows.
• Design and govern platform standards for memory lifecycle management, schema evolution, and metadata-driven registry integration.
• Collaborate with Science and Engineering teams to deliver end-to-end scalable solutions for cross-functional projects at organizational scale.
• Work closely with business and product stakeholders to define strategy, vision, and roadmap for platform-wide initiatives in personalization and recommendations.
• Set technical strategy for large-scale distributed systems that orchestrate workflows across memory stores, ML pipelines, and event-driven agents.
• Actively contribute to the research community and represent the company in technical and open-source forums.

What you will bring:

• You have expertise in architecting hybrid memory systems optimized for large-scale AI/ML systems.
• You have a proven track record of designing feature store platforms that serve both high-throughput real-time lookups and large-scale offline batch processing.
• You have advanced knowledge of distributed systems architecture.
• You have experience integrating ontologies and semantic search for memory retrieval in large-scale systems.
• You have hands-on experience in migrating or consolidating data/memory platforms at scale.
• You have strong influence skills for aligning cross-org initiatives and driving platform adoption.
• You have strong written and oral communication skills.

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 the future of retail.

Walmart’s culture is a competitive advantage, and it’s fostered by being together. Working together in person allows us to collaborate, align quickly, and innovate with greater speed. We use our campuses to create purposeful connections rooted in deepening understanding and investing in the development of our associates. Our hubs: Walmart is a global company with offices across the United States and around the world. Our global headquarters is in Bentonville, Arkansas, with primary hubs in the San Francisco Bay area and New York/New Jersey.

Benefits: Beyond our great compensation package, you can receive incentive awards for your performance. Other great perks include 401(k) match, stock purchase plan, paid maternity and parental leave, PTO, multiple health plans, and much more. 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 diversity- unique styles, experiences, identities, ideas and opinions – while being inclusive of all people.