Principal, Data Scientist

About the role

The Foundation Models team builds shared, reusable AI that the rest of Walmart builds on. We train models at scale, and they lift many products at the same time.

What you'll do

• This is a senior IC role with real room to set direction
• Define the problem statements and the solution direction
• We'll judge success by what ships and what customers and the business can measure, not by prototypes
• Identify problems: Find where foundation models can bring step-change gains across Walmart, then write the science vision and multi-quarter roadmap
• Influence: Build alignment across product, engineering, ads, merchandising, and supply chain teams in the US and India. Change long-held assumptions using evidence

What you need

• 10–18 yrs of experience
• Machine learning
• Python
• Deep learning
• LLMs
• Statistics
• PyTorch
• PhD or B.Tech or B.E. or B.Sc or BCA or M.Tech or MCA

Good to have

• Spark
• BigQuery
• IPR
• Prototyping