(USA) Staff, Software Engineer, Machine Learning

Position Summary...

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 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 ML model inferencing and governance at scale, contributing to highly engaging customer experiences across channels.

If you are someone who feels a personal stake in your work, loves data, enjoys solving complex problems, has a passion for privacy, and works well in a team, then this is the right place for you. Our team includes data scientists, engineers, and scientists passionate about using cutting-edge ML, deep learning, reinforcement learning, and NLP to deliver personalization and recommendation products.

About Team:

Our team collaborates closely with US stores and eCommerce to serve customers better through technological innovation. Walmart U.S. offers a wide range of products valued by customers shopping online or in-store, focusing on customers, stores, employees, merchant tools, data science, search, and personalization.

What you'll do:

• Lead development of feature store stacks supporting hybrid retrieval, diverse data modalities, and multi-tenant access control.
• Design and govern platform standards for memory management, schema evolution, and registry integration.
• Collaborate with science and engineering teams on scalable solutions for cross-functional projects.
• Work with stakeholders to define strategy, vision, and roadmap for personalization and recommendations.
• Set technical strategy for large-scale distributed systems managing workflows across various data and ML pipelines.
• Contribute to the research community and represent Walmart in technical forums.

What you'll bring:

• Expertise in architecting hybrid memory systems for large-scale AI/ML systems.
• Experience designing feature store platforms for real-time and offline processing.
• Advanced knowledge of distributed systems architecture.
• Experience with ontologies and semantic search in memory retrieval.
• Hands-on experience migrating or consolidating large-scale data platforms.
• Strong influence skills for cross-organization initiatives.
• Excellent communication skills.

About Walmart Global Tech

We innovate to make life easier for millions, within a collaborative, inclusive environment. We offer roles for all experience levels, focusing on growth, skills, and impact. Our culture emphasizes teamwork, innovation, and development, with offices worldwide, including Bentonville, AR, and hubs in the Bay Area and New York.

Benefits:

Competitive pay, performance bonuses, 401(k), stock plans, health coverage, paid leave, education benefits, and more. Live Better U offers fully paid education programs for associates.

Equal Opportunity Employer:

Walmart is committed to diversity and inclusion, respecting all identities and backgrounds. We strive to create an inclusive environment where everyone can thrive.