Staff, Software Engineer

Position Summary...

What you'll do...

About the Team & Role:

As a Staff Machine Learning Engineer, you will lead the technical vision for machine learning initiatives in Search & Personalization, driving innovation at scale. You will design cutting-edge ML models, lead engineering efforts, and mentor teams to push the boundaries of search relevance, query understanding,

and personalization strategies.

What You will Do:

• Architect and develop advanced ML models for real-time personalization and search ranking.

• Lead the design of scalable ML pipelines and infrastructure to support massive-scale search workloads.

• Drive research and experimentation in deep learning, reinforcement learning, and large-scale NLP.

• Collaborate with data science and engineering teams to translate research into production.

• Set technical direction for ML systems, ensuring robustness, scalability, and efficiency.

• Define best practices for model development, A/B testing, and continuous learning.

• Act as a mentor and technical leader, guiding ML engineers across teams.

What You Will Bring:

• 10+ years of experience in Machine Learning, Deep Learning, or related fields.

• Expertise in ML model architecture, feature engineering, and deployment.

• Strong proficiency in Python, TensorFlow/PyTorch, and large-scale data processing.

• Experience in production ML systems, distributed computing, and real-time inference.

• Proven leadership in ML projects, including setting technical direction and mentoring teams.

• Hands-on experience with cloud ML platforms (AWS/GCP/Azure).

Preferred Qualifications:

• PhD or Master's in Computer Science, Machine Learning, or a related field.

• Experience in Search, Ads, or Personalization at scale.

• Published research or patents in ML/AI.

About Walmart Global Tech

From entry-level to executive positions, Walmart provides limitless opportunities for growth, and career development. Walmart started small, with a single discount store and the simple philosophy of selling more for less. Today, we are a growing technology-enabled company founded on the same values as our first store. We establish clear expectations, empower associates to manage their work, and hold ourselves and one another to a high standard. Walmart's scale enables us to have an. No other company has the reach of Walmart, with 2.3 million associates worldwide and over 230 million weekly customers. Walmart is reshaping retail by investing in an expanding workforce. While technology is at the heart of our digital transformation, people are the reason we succeed and the force behind our innovations. We train our team in the skillsets of the future and bring in experts like you to help us grow.

Flexible, Hybrid Work

We use a hybrid way of working with primary in office presence coupled with an optimal mix of virtual presence. We use our campuses to collaborate and be together in person, as business needs require and for development and networking opportunities. This approach helps us make quicker decisions, remove location barriers across our global team, be more flexible in our personal lives.

Benefits

Beyond our great compensation package, you can receive incentive awards for your performance. Other great perks include a host of best-in-class benefits maternity and parental leave, PTO, health benefits, and much more.

Equal Opportunity Employer

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.

Minimum Qualifications...

Outlined below are the required minimum qualifications for this position. If none are listed, there are no minimum qualifications.

Preferred Qualifications...

Outlined below are the optional preferred qualifications for this position. If none are listed, there are no preferred qualifications.

Primary Location...